Skip to content

Refactor drjax primitives to use jnp.broadcast_to and improve explicit sharding support.#37

Merged
copybara-service[bot] merged 1 commit into
mainfrom
cl/820412444
Apr 3, 2026
Merged

Refactor drjax primitives to use jnp.broadcast_to and improve explicit sharding support.#37
copybara-service[bot] merged 1 commit into
mainfrom
cl/820412444

Refactor drjax primitives to use `jnp.broadcast_to` and improve expli…

8c609af
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar