google / aqtLinks
☆323Updated last month
Alternatives and similar repositories for aqt
Users that are interested in aqt are comparing it to the libraries listed below
Sorting:
- ☆187Updated 2 weeks ago
- This repository contains the experimental PyTorch native float8 training UX☆224Updated last year
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆517Updated last week
- jax-triton contains integrations between JAX and OpenAI Triton☆411Updated last month
- Fast low-bit matmul kernels in Triton☆338Updated this week
- extensible collectives library in triton☆88Updated 4 months ago
- A library for unit scaling in PyTorch☆128Updated 3 weeks ago
- Applied AI experiments and examples for PyTorch☆289Updated 2 months ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆563Updated 2 weeks ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆282Updated 3 weeks ago
- ☆171Updated last year
- A library to analyze PyTorch traces.☆398Updated last week
- Implementation of a Transformer, but completely in Triton☆273Updated 3 years ago
- ☆227Updated this week
- Implementation of Flash Attention in Jax☆215Updated last year
- A Quirky Assortment of CuTe Kernels☆374Updated this week
- JAX-Toolbox☆327Updated this week
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆258Updated last week
- Triton-based implementation of Sparse Mixture of Experts.☆227Updated 8 months ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆200Updated this week
- seqax = sequence modeling + JAX☆165Updated last week
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆207Updated last week
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆137Updated last year
- Accelerated First Order Parallel Associative Scan☆184Updated 11 months ago
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆366Updated last week
- Experiment of using Tangent to autodiff triton☆79Updated last year
- An open-source efficient deep learning framework/compiler, written in python.☆710Updated last week
- ☆251Updated last year
- FlashFFTConv: Efficient Convolutions for Long Sequences with Tensor Cores☆325Updated 7 months ago
- ☆142Updated 2 weeks ago