jxbz / agdLinks
Automatic gradient descent
☆217Updated 2 years ago
Alternatives and similar repositories for agd
Users that are interested in agd are comparing it to the libraries listed below
Sorting:
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆298Updated last year
- ☆118Updated last month
- ☆250Updated 6 months ago
- LoRA for arbitrary JAX models and functions☆143Updated last year
- Neural Networks for JAX☆84Updated last year
- ☆233Updated 11 months ago
- Named tensors with first-class dimensions for PyTorch☆332Updated 2 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆242Updated 2 years ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆406Updated this week
- Unofficial JAX implementations of deep learning research papers☆160Updated 3 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆190Updated 3 years ago
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.☆174Updated 2 years ago
- A functional training loops library for JAX☆88Updated last year
- git extension for {collaborative, communal, continual} model development☆217Updated last year
- An interactive exploration of Transformer programming.☆271Updated 2 years ago
- JAX Synergistic Memory Inspector☆184Updated last year
- Train very large language models in Jax.☆210Updated 2 years ago
- ☆288Updated last year
- Graph neural networks in JAX.☆68Updated last year
- Running Jax in PyTorch Lightning☆118Updated last year
- A simple library for scaling up JAX programs☆144Updated 2 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆191Updated last week
- JMP is a Mixed Precision library for JAX.☆211Updated 11 months ago
- Hierarchical Associative Memory User Experience☆106Updated this week
- Functional local implementations of main model parallelism approaches☆95Updated 2 years ago
- Compositional Linear Algebra☆505Updated 5 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆110Updated last year
- Run PyTorch in JAX. 🤝☆311Updated 3 months ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆181Updated 8 months ago
- JAX implementation of the Llama 2 model☆215Updated last year