paxml
ML framework
A framework for configuring and running machine learning experiments on top of Jax.
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.
457 stars
15 watching
68 forks
Language: Python
last commit: 11 days ago
Linked from 1 awesome list
c4gptjaxlarge-language-modelsllmmodel-flopsparallelism
n2cholas/awesome-jaxRelated projects:
| Repository | Description | Stars |
|---|---|---|
| google/praxis | A library designed to be usable by other JAX-based ML projects, providing optimized layers for machine learning at scale. | 178 |
 gomlx/gomlx | A high-performance machine learning framework built on just-in-time compilation and automatic differentiation. | 354 |
 erfanzar/easydel | A flexible framework for training and serving machine learning models with JAX/Flax | 206 |
 symjax/symjax | A symbolic programming library for machine learning and deep learning applications. | 120 |
 ai-hypercomputer/maxtext | A high-performance LLM written in Python/Jax for training and inference on Google Cloud TPUs and GPUs. | 1,529 |
 cdipaolo/goml | A Go library that provides an online machine learning framework with various algorithms and models | 1,568 |
 google-deepmind/jaxline | Provides a Python-based framework for building distributed JAX training and evaluation experiments | 152 |
 matthias-wright/flaxmodels | Provides pre-trained deep learning models for the Jax/Flax ecosystem. | 238 |
 jaxgaussianprocesses/gpjax | Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation | 461 |
 microsoft/0xdeca10b | 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. | 556 |
 scicloj/scicloj.ml.clj-djl | Provides pre-trained machine learning models for natural language processing tasks using Clojure and the clj-djl framework. | 0 |
| google/objax | An open-source machine learning framework designed to simplify and accelerate research | 771 |
 rlouf/mcx | Tools and methods for Bayesian deep learning using probabilistic programming. | 325 |
 neuraxio/neuraxle | A machine learning pipeline library that enables the creation of modular and reusable data processing workflows | 608 |
 yandex/rep | A toolset for building and running reproducible machine learning experiments in Python | 689 |