search

proteneer / timemachine

Differentiate all the things!
Other
140 stars 17 forks source link

Update jax to 0.4.28 to fix memory corruption issue #1388

Closed jkausrelay closed 2 months ago

jkausrelay commented 2 months ago

See https://github.com/jax-ml/jax/releases/tag/jaxlib-v0.4.28