Igor Kuszczak comments

Repositories
Issues
Comments

Results 12 comments of


                                            Igor Kuszczak

Enabling tracer leak check causes error

I tried the code as well and managed to reproduce the error. I think it might be related to the following issue in JAX: https://github.com/jax-ml/jax/issues/26560

[QUESTION] Optimization example with JAX

Great, looking forward to this!