albanD / subclass_zooLinks
☆167Updated 11 months ago
Alternatives and similar repositories for subclass_zoo
Users that are interested in subclass_zoo are comparing it to the libraries listed below
Sorting:
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆150Updated this week
- extensible collectives library in triton☆87Updated 2 months ago
- ☆208Updated 10 months ago
- Applied AI experiments and examples for PyTorch☆271Updated this week
- ☆86Updated 5 months ago
- ☆210Updated this week
- A library to analyze PyTorch traces.☆379Updated last week
- This repository contains the experimental PyTorch native float8 training UX☆223Updated 10 months ago
- Cataloging released Triton kernels.☆226Updated 4 months ago
- Collection of kernels written in Triton language☆125Updated last month
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆127Updated this week
- Shared Middle-Layer for Triton Compilation☆251Updated this week
- ☆79Updated 6 months ago
- Fast low-bit matmul kernels in Triton☆303Updated last week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆329Updated this week
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆157Updated 5 months ago
- Research and development for optimizing transformers☆126Updated 4 years ago
- ☆310Updated last week
- Github mirror of trition-lang/triton repo.☆37Updated this week
- Training neural networks in TensorFlow 2.0 with 5x less memory☆131Updated 3 years ago
- MLIR-based partitioning system☆86Updated this week
- jax-triton contains integrations between JAX and OpenAI Triton☆395Updated this week
- Fastest kernels written from scratch☆269Updated 2 months ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆196Updated this week
- PyTorch RFCs (experimental)☆132Updated last week
- Implementation of a Transformer, but completely in Triton☆265Updated 3 years ago
- ☆105Updated 9 months ago
- A library for unit scaling in PyTorch☆125Updated 6 months ago
- Convert nvprof profiles into about:tracing compatible JSON files☆69Updated 4 years ago
- The simplest but fast implementation of matrix multiplication in CUDA.☆35Updated 10 months ago