young-geng/scalax

A simple library for scaling up JAX programs

1 best points · 0 comments · 1 Hacker News post · 149 GitHub stars · View on GitHub

Hacker News discussions

• Scalax: A simple library for scaling up Jax programs (1 points, 0 comments, by lnyan, 2024-09-01)