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.
created at June 14, 2022, 4:04 p.m.
gNXI Tools - gRPC Network Management/Operations Interface Tools
created at Sept. 26, 2017, 8:19 a.m.
FedJAX is a JAX-based open source library for Federated Learning simulations that emphasizes ease-of-use in research.
created at Dec. 22, 2020, 2:44 p.m.
Mathematical operations for JAX pytrees
created at Dec. 18, 2021, 1:56 a.m.
Automatically exported from code.google.com/p/closure-linter
created at June 17, 2015, 6:34 a.m.