-
-
paxml Public
Forked from google/paxmlPax 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 lead…
-
flax Public
Forked from google/flaxFlax is a neural network library for JAX that is designed for flexibility.
-
maxtext Public
Forked from AI-Hypercomputer/maxtextA simple, performant and scalable Jax LLM!
Python Apache License 2.0 UpdatedSep 30, 2024 -
-
xla Public
Forked from openxla/xlaA machine learning compiler for GPUs, CPUs, and ML accelerators
C++ Apache License 2.0 UpdatedApr 17, 2024 -