vdesai2014 / inference-optimization-blog-post
☆86Updated 11 months ago
Alternatives and similar repositories for inference-optimization-blog-post:
Users that are interested in inference-optimization-blog-post are comparing it to the libraries listed below
- ☆142Updated last year
- ☆75Updated 7 months ago
- ring-attention experiments☆123Updated 4 months ago
- This repository contains the experimental PyTorch native float8 training UX☆221Updated 6 months ago
- seqax = sequence modeling + JAX☆143Updated 7 months ago
- Experiment of using Tangent to autodiff triton☆75Updated last year
- KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems☆183Updated this week
- ☆179Updated last week
- JAX bindings for Flash Attention v2☆85Updated 7 months ago
- Normalized Transformer (nGPT)☆152Updated 3 months ago
- ☆53Updated last year
- Cataloging released Triton kernels.☆168Updated last month
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆116Updated 2 months ago
- Fast Matrix Multiplications for Lookup Table-Quantized LLMs☆229Updated this week
- Solve puzzles. Learn CUDA.☆62Updated last year
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆222Updated this week
- supporting pytorch FSDP for optimizers☆76Updated 2 months ago
- ☆135Updated last week
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆118Updated last year
- A library for unit scaling in PyTorch☆122Updated 2 months ago
- Fast low-bit matmul kernels in Triton☆236Updated this week
- Applied AI experiments and examples for PyTorch☆225Updated this week
- Write a fast kernel and run it on Discord. See how you compare against the best!☆18Updated this week
- A MAD laboratory to improve AI architecture designs 🧪☆102Updated 2 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆122Updated 10 months ago
- Collection of autoregressive model implementation☆81Updated this week
- Understand and test language model architectures on synthetic tasks.☆181Updated last month
- Collection of kernels written in Triton language☆105Updated this week
- Implementation of Diffusion Transformer (DiT) in JAX☆265Updated 8 months ago
- Accelerated First Order Parallel Associative Scan☆171Updated 6 months ago