google-deepmind / nanodoLinks
☆285Updated last year
Alternatives and similar repositories for nanodo
Users that are interested in nanodo are comparing it to the libraries listed below
Sorting:
- 🧱 Modula software package☆316Updated 4 months ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆687Updated 3 weeks ago
- seqax = sequence modeling + JAX☆169Updated 5 months ago
- Minimal yet performant LLM examples in pure JAX☆219Updated 2 weeks ago
- A simple library for scaling up JAX programs☆144Updated last month
- JAX Synergistic Memory Inspector☆183Updated last year
- LoRA for arbitrary JAX models and functions☆143Updated last year
- Named Tensors for Legible Deep Learning in JAX☆215Updated last month
- Efficient optimizers☆277Updated this week
- Minimal but scalable implementation of large language models in JAX☆35Updated 3 weeks ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆180Updated 5 months ago
- Cost aware hyperparameter tuning algorithm☆176Updated last year
- A MAD laboratory to improve AI architecture designs 🧪☆135Updated last year
- Accelerate, Optimize performance with streamlined training and serving options with JAX.☆325Updated last week
- Implementation of PSGD optimizer in JAX☆35Updated 11 months ago
- ☆234Updated 10 months ago
- Jax/Flax rewrite of Karpathy's nanoGPT☆62Updated 2 years ago
- Implementation of Diffusion Transformer (DiT) in JAX☆298Updated last year
- Understand and test language model architectures on synthetic tasks.☆246Updated 2 months ago
- Distributed pretraining of large language models (LLMs) on cloud TPU slices, with Jax and Equinox.☆24Updated last year
- Accelerated First Order Parallel Associative Scan☆193Updated last year
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆335Updated last month
- ☆91Updated last year
- jax-triton contains integrations between JAX and OpenAI Triton☆436Updated last week
- supporting pytorch FSDP for optimizers☆84Updated last year
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆404Updated this week
- ☆460Updated last year
- Dion optimizer algorithm☆404Updated last week
- Library for reading and processing ML training data.☆633Updated this week
- ☆229Updated last year