google / praxisLinks
☆188Updated last week
Alternatives and similar repositories for praxis
Users that are interested in praxis are comparing it to the libraries listed below
Sorting:
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆536Updated last month
- Implementation of Flash Attention in Jax☆218Updated last year
- jax-triton contains integrations between JAX and OpenAI Triton☆424Updated 3 weeks ago
- ☆331Updated 3 weeks ago
- ☆362Updated last year
- ☆146Updated last week
- JAX Synergistic Memory Inspector☆180Updated last year
- JMP is a Mixed Precision library for JAX.☆208Updated 8 months ago
- Train very large language models in Jax.☆209Updated last year
- JAX implementation of the Llama 2 model☆218Updated last year
- Implementation of a Transformer, but completely in Triton☆275Updated 3 years ago
- LoRA for arbitrary JAX models and functions☆142Updated last year
- Inference code for LLaMA models in JAX☆120Updated last year
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆123Updated 2 weeks ago
- JAX-Toolbox☆343Updated this week
- A simple library for scaling up JAX programs☆143Updated 11 months ago
- seqax = sequence modeling + JAX☆167Updated 2 months ago
- ☆233Updated 7 months ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆189Updated 3 years ago
- A library for unit scaling in PyTorch☆130Updated 2 months ago
- Library for reading and processing ML training data.☆548Updated this week
- Orbax provides common checkpointing and persistence utilities for JAX users☆428Updated this week
- Minimal yet performant LLM examples in pure JAX☆177Updated last week
- ☆62Updated 3 years ago
- ☆281Updated last year
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆667Updated this week
- If it quacks like a tensor...☆59Updated 10 months ago
- ☆115Updated 3 weeks ago
- CLU lets you write beautiful training loops in JAX.☆356Updated 3 months ago
- Two implementations of ZeRO-1 optimizer sharding in JAX☆14Updated 2 years ago