-
Notifications
You must be signed in to change notification settings - Fork 37
Pull requests: google-deepmind/torax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Cleanup of source model function documentation.
#629
opened Dec 24, 2024 by
copybara-service
bot
Loading…
Optimize array stacking in the output by using NumPy rather than JAX.
#586
opened Dec 2, 2024 by
copybara-service
bot
Loading…
Add support for trace checking in jax_utils.jit.
#359
opened Aug 20, 2024 by
copybara-service
bot
Loading…
ProTip!
Add no:assignee to see everything that’s not assigned.