Build compute kernels and load them from the Hub.

Developer Asset Hub for NVIDIA Nemotron — A one-stop resource for training recipes, usage cookbooks, datasets, and full end-to-end reference examples to build with Nemotron models

Python Toolkit for Causal and Probabilistic Reasoning

Puffing up reinforcement learning

Any model. Any hardware. Zero compromise. Built with @ziglang / @openxla / MLIR / @bazelbuild

HRM-Text is a 1B text generation model based on the HRM architecture, strengthened by task completion and latent space reasoning.

[ICLR2025, ICML2025, NeurIPS2025 Spotlight] Quantized Attention achieves speedup of 2-5x compared to FlashAttention, without losing end-to-end metrics across language, image, and video models.

Fast and memory-efficient exact attention

Muon is an optimizer for hidden layers in neural networks

Ultra fast asyncio event loop.

The little ASGI framework that shines. 🌟

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

An extremely fast Python type checker and language server, written in Rust.

A conda-forge distribution.

Implementation of the sap-rpt-1-oss deep learning model with inference pipeline as described in the paper "ConTextTab: A Semantics-Aware Tabular In-Context Learner".

A game theoretic approach to explain the output of any machine learning model.

Hierarchical Reasoning Model Official Release

DSPy: The framework for programming—not prompting—language models

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

An implementation of Performer, a linear attention-based transformer, in Pytorch

Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch

250+ Fine-tuning & RL Notebooks for text, vision, audio, embedding, TTS models.

Extremely fast Query Engine for DataFrames, written in Rust

Machine learning with dataframes

LLM training in simple, raw C/CUDA

Implementation of Self-Compressing Neural Networks in Pytorch https://arxiv.org/abs/2301.13142

⏰ AI conference deadline countdowns

A fast, scalable, high performance Gradient Boosting on Decision Trees library, used for ranking, classification, regression and other machine learning tasks for Python, R, Java, C++. Supports comp…