A debugging and profiling tool that can trace and visualize python code execution

RND1: Scaling Diffusion Language Models

[NeurIPS 2025 Oral] Exploring Diffusion Transformer Designs via Grafting

A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.

CUDA Core Compute Libraries

Genome modeling and design across all domains of life

Small scale distributed training of sequential deep learning models, built on Numpy and MPI.

common in-memory tensor structure

Mirage Persistent Kernel: Compiling LLMs into a MegaKernel

TUI for the Slurm Workload Manager

simple bibtex generator for any text with \cite{}

High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.

A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.

Distilabel is a framework for synthetic data and AI feedback for engineers who need fast, reliable and scalable pipelines based on verified research papers.

Arena-Hard-Auto: An automatic LLM benchmark.

A MAD laboratory to improve AI architecture designs 🧪

Official repo for "Mini-Gemini: Mining the Potential of Multi-modality Vision Language Models"

Fast Multi-dimensional Sparse Attention

Fine-tuning & Reinforcement Learning for LLMs. 🦥 Train OpenAI gpt-oss, DeepSeek-R1, Qwen3, Gemma 3, TTS 2x faster with 70% less VRAM.

Large Context Attention

[ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)

Biological foundation modeling from molecular to genome scale

Memray is a memory profiler for Python

A family of lightweight multimodal models.

FlashInfer: Kernel Library for LLM Serving

Minimalistic large language model 3D-parallelism training