Reviews
Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.
Search similar apps
License
Apache License 2.0
Related apps
angle
A conformant OpenGL ES implementation for Windows, Mac, Linux, iOS and Android.
C++3116other
10 months ago
argh
Rust derive-based argument parsing optimized for code size
Rust1533bsd-3-clause
3 months ago
arghargument-parserarguments
brax
Massively parallel rigidbody physics simulation on accelerator hardware.
Jupyter Notebook1932apache-2.0
4 months ago
jaxphysics-simulationreinforcement-learning
cadvisor
Analyzes resource usage and performance characteristics of running containers.
Go15892other
4 months ago
cpu_features
A cross platform C99 library to get cpu features at runtime.
C++2361apache-2.0
2 months ago
cpucpu-featurescross-platform
error-prone
Catch common Java mistakes as compile-time errors
Java6666apache-2.0
2 months ago
javastatic-analysis