Geeks
young-geng
/
scalax
A simple library for scaling up JAX programs
Primary Language
Python
Apache License 2.0
Apache-2.0
scalax GitHub
Markdown
Readme
2
Issues
134
Stargazers
8
Watchers
Watchers
eedalong
Alibaba PAI Team
eemailme
JiahaoYao
University of California, Berkeley
philippe-eecs
UT Austin
Sea-Snell
Berkeley, CA
sodabeta7
Apple
wookayin
University of Michigan
young-geng
Google DeepMind