8000 [WIP] Update to jax 0.6, support arbitrary sharding (drop MPI) by PhilipVinc · Pull Request #2059 · netket/netket · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

[WIP] Update to jax 0.6, support arbitrary sharding (drop MPI)#2059

Draft
PhilipVinc wants to merge 40 commits intomasterfrom
pv/jax-0.6
Draft

[WIP] Update to jax 0.6, support arbitrary sharding (drop MPI)#2059
PhilipVinc wants to merge 40 commits intomasterfrom
pv/jax-0.6

Commits

Commits on May 31, 2025

Commits on Jun 3, 2025

Commits on Jun 4, 2025

Commits on Jun 6, 2025

Commits on Jun 20, 2025

Commits on Jun 26, 2025

Commits on Jun 27, 2025
0