-
Notifications
You must be signed in to change notification settings - Fork 1
Pull requests: google/drjax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Make drjax fully compatible with JAX Explicit Sharding.
#39
opened May 15, 2026 by
copybara-service
Bot
Loading…
Finalize deprecation of several
jax.core APIs for JAX v0.10.0
#35
opened Apr 2, 2026 by
copybara-service
Bot
Loading…
Ensure primitives propagate sharding annotations in abstract evaluation.
#33
opened Oct 20, 2025 by
copybara-service
Bot
Loading…
Bug Fix: Tracers do not always shave
.sharding, use jax.typeof to get an abstract value.
#28
opened Oct 4, 2025 by
copybara-service
Bot
Loading…
ProTip!
Find all pull requests that aren't related to any open issues with -linked:issue.