paxml
by google
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.
AI summary
ML framework
A framework for configuring and running machine learning experiments on top of Jax.
- stars
- 461
- forks
- 69
- watching
- 15
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
ML framework component
A library designed to be usable by other JAX-based ML projects, providing optimized layers for machine learning at scale.
gomlx/gomlx553
Machine learning framework
An accelerated machine learning framework for Go that provides tools and libraries for training, fine-tuning, and modifying machine learning models.
ML framework builder
A framework for training and serving machine learning models using JAX, with a focus on flexibility and customization.
ML framework
A symbolic programming library for machine learning and deep learning applications.
LLM framework
A high-performance LLM written in Python/Jax for training and inference on Google Cloud TPUs and GPUs.
cdipaolo/goml1.6K
ML framework
A Go library that provides an online machine learning framework with various algorithms and models
Experiment framework
Provides a Python-based framework for building distributed JAX training and evaluation experiments
Model repository
Provides pre-trained deep learning models for the Jax/Flax ecosystem.
GP model library
Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation
Machine Learning Framework
A framework for hosting and training machine learning models on a blockchain, enabling secure sharing and prediction without requiring users to pay for data or model updates.
ML models
Provides pre-trained machine learning models for natural language processing tasks using Clojure and the clj-djl framework.
Research accelerator
An open-source machine learning framework designed to simplify and accelerate research
rlouf/mcx328
Probabilistic programming toolset
Tools and methods for Bayesian deep learning using probabilistic programming.
Pipeline framework
A machine learning pipeline library that enables the creation of modular and reusable data processing workflows
yandex/rep689
ML lab
A toolset for building and running reproducible machine learning experiments in Python
Featured in 1 awesome list
Each link jumps to the spot where the list mentions paxml.
