Skip to content

Make drjax fully compatible with JAX Explicit Sharding.#39

Merged
copybara-service[bot] merged 1 commit into
mainfrom
cl/914948662
Jun 12, 2026
Merged

Make drjax fully compatible with JAX Explicit Sharding.#39
copybara-service[bot] merged 1 commit into
mainfrom
cl/914948662

Make drjax fully compatible with JAX Explicit Sharding.

8c5894f
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar