marin-community / levanterLinks
Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
β675Updated this week
Alternatives and similar repositories for levanter
Users that are interested in levanter are comparing it to the libraries listed below
Sorting:
- β283Updated last year
- π§± Modula software packageβ291Updated 2 months ago
- Accelerate, Optimize performance with streamlined training and serving options with JAX.β317Updated this week
- JAX Synergistic Memory Inspectorβ179Updated last year
- Named Tensors for Legible Deep Learning in JAXβ211Updated last week
- Library for reading and processing ML training data.β581Updated this week
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentaβ¦β539Updated last month
- jax-triton contains integrations between JAX and OpenAI Tritonβ428Updated 2 weeks ago
- Minimal yet performant LLM examples in pure JAXβ187Updated last month
- seqax = sequence modeling + JAXβ168Updated 3 months ago
- Puzzles for exploring transformersβ376Updated 2 years ago
- Orbax provides common checkpointing and persistence utilities for JAX usersβ439Updated this week
- JAX-Toolboxβ355Updated last week
- β542Updated last year
- JAX implementation of the Llama 2 modelβ216Updated last year
- Open-source framework for the research and development of foundation models.β532Updated this week
- β456Updated last year
- β190Updated last week
- CLU lets you write beautiful training loops in JAX.β358Updated 4 months ago
- Inference code for LLaMA models in JAXβ119Updated last year
- For optimization algorithm research and development.β543Updated last week
- Annotated version of the Mamba paperβ490Updated last year
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.β295Updated last year
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvementβ¦β400Updated last week
- Efficient optimizersβ275Updated 2 weeks ago
- LoRA for arbitrary JAX models and functionsβ141Updated last year
- β362Updated last year
- β335Updated last month
- A simple library for scaling up JAX programsβ144Updated 11 months ago
- Everything you want to know about Google Cloud TPUβ549Updated last year