axonn-ai / axonnLinks
Parallel framework for training and fine-tuning deep neural networks
☆70Updated 2 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☆93Updated 9 months ago
- Triton-based Symmetric Memory operators and examples☆79Updated last week
- Ship correct and fast LLM kernels to PyTorch☆135Updated last week
- A bunch of kernels that might make stuff slower 😉☆75Updated this week
- ring-attention experiments☆161Updated last year
- ☆115Updated last year
- Write a fast kernel and run it on Discord. See how you compare against the best!☆67Updated this week
- This repository contains the experimental PyTorch native float8 training UX☆227Updated last year
- A Python library transfers PyTorch tensors between CPU and NVMe☆125Updated last year
- Collection of kernels written in Triton language☆174Updated 9 months ago
- Triton-based implementation of Sparse Mixture of Experts.☆260Updated 3 months ago
- ☆71Updated 9 months ago
- Framework to reduce autotune overhead to zero for well known deployments.☆92Updated 4 months ago
- ☆28Updated last year
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆218Updated this week
- PyTorch bindings for CUTLASS grouped GEMM.☆140Updated 7 months ago
- Hydragen: High-Throughput LLM Inference with Shared Prefixes☆46Updated last year
- Hand-Rolled GPU communications library☆76Updated last month
- Evaluating Large Language Models for CUDA Code Generation ComputeEval is a framework designed to generate and evaluate CUDA code from Lar…☆92Updated 2 weeks ago
- Applied AI experiments and examples for PyTorch☆314Updated 5 months ago
- ☆101Updated last year
- Automatic differentiation for Triton Kernels☆29Updated 5 months ago
- Boosting 4-bit inference kernels with 2:4 Sparsity☆91Updated last year
- High-Performance FP32 GEMM on CUDA devices☆117Updated last year
- ☆39Updated last month
- QuTLASS: CUTLASS-Powered Quantized BLAS for Deep Learning☆161Updated 2 months ago
- Experiment of using Tangent to autodiff triton☆81Updated 2 years ago
- ☆15Updated 2 months ago
- ☆273Updated this week
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆278Updated last month