Karen Chung comments

Repositories
Issues
Comments

Results 1 comments of


                                            Karen Chung

Jax leaks memory with random keys

Any updates to this issue @yashk2810 ? I'm on jax==0.4.23 and jaxlib==0.4.23+cuda11.cudnn86 (which should include changes from #17452 ) and see the same memory leak issue.