-
Notifications
You must be signed in to change notification settings - Fork 2.9k
Pull requests: jax-ml/jax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
DOC: Remove duplicate
JAX version 0.4.37
heading in Changelog
#26840
opened Feb 28, 2025 by
rajasekharporeddy
Fixed printing order of results in
jax.debug.print
documentation
#26839
opened Feb 28, 2025 by
Sai-Suraj-27
Plumb layout through the creation of IFRT Arrays (roll-forward with fix).
#26838
opened Feb 28, 2025 by
copybara-service
bot
Add build targets for
jax-rocm-plugin
and jax-rocm-pjrt
wheels.
#26827
opened Feb 27, 2025 by
copybara-service
bot
Print the list of installed packages before running pytests
#26826
opened Feb 27, 2025 by
copybara-service
bot
Reimplement Ready for copybara import and testing
custom_vjp.optimize_remat
using custom_dce
pull ready
Update ad.backward_pass to support non-linear functions of constants
#26811
opened Feb 27, 2025 by
dfm
Fix assertion fails on DebugInfo with JAX_USE_DIRECT_LINEARIZE=1
#26810
opened Feb 27, 2025 by
gnecula
[Mosaic GPU] Add support for small RHS tile sizes in WGMMA
#26807
opened Feb 27, 2025 by
copybara-service
bot
Remove
tensorstore
dependency from test dependencies for python3.13-nogil
jobs.
#26777
opened Feb 26, 2025 by
copybara-service
bot
[Pallas/Mosaic GPU] Add an abstraction to obtain a slice of dynamic shared memory when using waprgroup semantics.
#26773
opened Feb 26, 2025 by
copybara-service
bot
Add functionality to allow promoting RC wheels during release
#26756
opened Feb 26, 2025 by
copybara-service
bot
[Mosaic:TPU][Relayout] Row shifts for packed types and non-native tilings
#26754
opened Feb 25, 2025 by
copybara-service
bot
Debug failing CUDA test jobs
CI Connection Halt - On Retry
Flag every job that has its workflow file setup for halting to halt if the workflow is retried
#26749
opened Feb 25, 2025 by
copybara-service
bot
For testing jaxlib at head, install the libtpu version that
jax
pins
#26709
opened Feb 24, 2025 by
copybara-service
bot
Use the same name for aliased Vars when pretty-printing Jaxprs.
pull ready
Ready for copybara import and testing
#26697
opened Feb 24, 2025 by
gnecula
[Pallas] Add support for setting operand as HBM memory.
#26694
opened Feb 24, 2025 by
copybara-service
bot
Previous Next
ProTip!
Filter pull requests by the default branch with base:main.