-
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
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!
Mix and match filters to narrow down what you’re looking for.