albanD / subclass_zoo
☆148Updated 5 months ago
Related projects ⓘ
Alternatives and complementary repositories for subclass_zoo
- extensible collectives library in triton☆71Updated last month
- ☆167Updated 4 months ago
- A library to analyze PyTorch traces.☆307Updated this week
- Research and development for optimizing transformers☆125Updated 3 years ago
- ☆152Updated this week
- A library of GPU kernels for sparse matrix operations.☆249Updated 3 years ago
- ☆156Updated last year
- ☆267Updated this week
- This repository contains the experimental PyTorch native float8 training UX☆211Updated 3 months ago
- Cataloging released Triton kernels.☆134Updated 2 months ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆270Updated this week
- Applied AI experiments and examples for PyTorch☆166Updated 3 weeks ago
- ☆45Updated 2 weeks ago
- jax-triton contains integrations between JAX and OpenAI Triton☆343Updated 3 weeks ago
- ☆140Updated last year
- A schedule language for large model training☆141Updated 5 months ago
- Shared Middle-Layer for Triton Compilation☆191Updated this week
- Experimental projects related to TensorRT☆81Updated this week
- Benchmark code for the "Online normalizer calculation for softmax" paper☆59Updated 6 years ago
- A tensor-aware point-to-point communication primitive for machine learning☆249Updated last year
- ☆48Updated 8 months ago
- A library for unit scaling in PyTorch☆105Updated 2 weeks ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆146Updated 2 weeks ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆90Updated 4 months ago
- Stores documents and resources used by the OpenXLA developer community☆107Updated 3 months ago
- Simple and fast low-bit matmul kernels in CUDA / Triton☆143Updated this week
- Memory Optimizations for Deep Learning (ICML 2023)☆60Updated 8 months ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆165Updated this week
- Training neural networks in TensorFlow 2.0 with 5x less memory☆129Updated 2 years ago
- ☆48Updated this week