drisspg / driss_torchLinks
Cuda extensions for PyTorch
☆12Updated last month
Alternatives and similar repositories for driss_torch
Users that are interested in driss_torch are comparing it to the libraries listed below
Sorting:
- ☆21Updated 10 months ago
- Experiment of using Tangent to autodiff triton☆81Updated last year
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆154Updated 2 years ago
- Nsight Python is a Python kernel profiling interface based on NVIDIA Nsight Tools☆89Updated last week
- High-Performance SGEMM on CUDA devices☆115Updated 11 months ago
- PTX-Tutorial Written Purely By AIs (Deep Research of Openai and Claude 3.7)☆66Updated 9 months ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆325Updated last week
- train with kittens!☆63Updated last year
- ☆91Updated last year
- LLM training in simple, raw C/CUDA☆110Updated last year
- extensible collectives library in triton☆92Updated 9 months ago
- ☆124Updated last year
- ☆178Updated last year
- ☆19Updated last month
- ☆342Updated last week
- Write a fast kernel and run it on Discord. See how you compare against the best!☆66Updated last week
- ring-attention experiments☆161Updated last year
- ☆12Updated last year
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆130Updated 3 weeks ago
- MoE training for Me and You and maybe other people☆319Updated 2 weeks ago
- Accelerated First Order Parallel Associative Scan☆193Updated last week
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆195Updated 7 months ago
- This repository contains the experimental PyTorch native float8 training UX☆227Updated last year
- Learn CUDA with PyTorch☆176Updated 3 weeks ago
- Ship correct and fast LLM kernels to PyTorch☆132Updated this week
- Collection of kernels written in Triton language☆174Updated 9 months ago
- ☆92Updated last year
- A curated collection of resources, tutorials, and best practices for learning and mastering NVIDIA CUTLASS☆247Updated 8 months ago
- CUDA and Triton implementations of Flash Attention with SoftmaxN.☆73Updated last year
- Various transformers for FSDP research☆38Updated 3 years ago