google-research / jaxpruner
☆221Updated last month
Alternatives and similar repositories for jaxpruner:
Users that are interested in jaxpruner are comparing it to the libraries listed below
- A simple library for scaling up JAX programs☆134Updated 5 months ago
- A library for unit scaling in PyTorch☆125Updated 4 months ago
- JMP is a Mixed Precision library for JAX.☆193Updated 2 months ago
- LoRA for arbitrary JAX models and functions☆135Updated last year
- jax-triton contains integrations between JAX and OpenAI Triton☆388Updated this week
- ☆215Updated 8 months ago
- JAX Synergistic Memory Inspector☆171Updated 8 months ago
- ☆113Updated this week
- 🧱 Modula software package☆187Updated this week
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆185Updated 3 months ago
- Train very large language models in Jax.☆203Updated last year
- CLU lets you write beautiful training loops in JAX.☆335Updated 3 weeks ago
- Accelerated First Order Parallel Associative Scan☆180Updated 7 months ago
- seqax = sequence modeling + JAX☆151Updated 2 weeks ago
- Experiment of using Tangent to autodiff triton☆78Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆171Updated this week
- ☆52Updated 6 months ago
- A MAD laboratory to improve AI architecture designs 🧪☆108Updated 3 months ago
- ☆202Updated 2 years ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆372Updated this week
- Automatic gradient descent☆207Updated last year
- OpTree: Optimized PyTree Utilities☆175Updated last week
- Unofficial JAX implementations of deep learning research papers☆154Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- JAX bindings for Flash Attention v2☆89Updated 8 months ago
- Named Tensors for Legible Deep Learning in JAX☆169Updated this week
- ☆154Updated last year
- ☆147Updated this week
- Run PyTorch in JAX. 🤝☆232Updated last month
- ☆185Updated last week