axonn-ai / axonnLinks
A parallel framework for training deep neural networks
β63Updated 5 months ago
Alternatives and similar repositories for axonn
Users that are interested in axonn are comparing it to the libraries listed below
Sorting:
- extensible collectives library in tritonβ88Updated 4 months ago
- A bunch of kernels that might make stuff slower πβ58Updated last week
- This repository contains the experimental PyTorch native float8 training UXβ224Updated last year
- Collection of kernels written in Triton languageβ147Updated 4 months ago
- Experiment of using Tangent to autodiff tritonβ80Updated last year
- β110Updated last year
- β232Updated last week
- ring-attention experimentsβ149Updated 10 months ago
- Memory Optimizations for Deep Learning (ICML 2023)β102Updated last year
- β88Updated 9 months ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.β271Updated this week
- π Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.β208Updated last week
- Write a fast kernel and run it on Discord. See how you compare against the best!β51Updated this week
- Cataloging released Triton kernels.β252Updated 7 months ago
- Applied AI experiments and examples for PyTorchβ291Updated last week
- Sparsity support for PyTorchβ36Updated 5 months ago
- A Python library transfers PyTorch tensors between CPU and NVMeβ120Updated 9 months ago
- β118Updated last year
- Effective transpose on Hopper GPUβ23Updated 3 months ago
- β28Updated 7 months ago
- Triton-based implementation of Sparse Mixture of Experts.β233Updated this week
- β74Updated 5 months ago
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.β136Updated last year
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mindβ¦β159Updated 2 months ago
- train with kittens!β62Updated 10 months ago
- PyTorch bindings for CUTLASS grouped GEMM.β110Updated 3 months ago
- Triton-based Symmetric Memory operators and examplesβ23Updated this week
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.β207Updated this week
- Boosting 4-bit inference kernels with 2:4 Sparsityβ80Updated 11 months ago
- Fast low-bit matmul kernels in Tritonβ353Updated last week