gpt-oss-120b and gpt-oss-20b are two open-weight language models by OpenAI

TPU inference for vLLM, with unified JAX and PyTorch support.

This package contains the original 2012 AlexNet code.

A JAX-based simulator for autonomous driving research.

Convolutional Neural Networks

The Modular Platform (includes MAX & Mojo)

Merlion: A Machine Learning Framework for Time Series Intelligence

My CS231n lecture notes

A machine learning compiler for GPUs, CPUs, and ML accelerators

Backward compatible ML compute opset inspired by HLO/MHLO

Tracking RISC-V Actions on Education, Training, Courses, Monitorships, etc.

[WIP] Resources for AI engineers. Also contains supporting materials for the book AI Engineering (Chip Huyen, 2025)

Everything we actually know about the Apple Neural Engine (ANE)

Vector math library using RISC-V vector ISA via C intrinsic

PROSE is a tool for automated floating-point precision tuning of Fortran code designed to target large weather and climate models.

Samples for CUDA Developers which demonstrates features in CUDA Toolkit

A highly-flexible GPU simulator for AMD GPUs.

[DEPRECATED] Moved to ROCm/rocm-libraries repo. NOTE: develop branch is maintained as a read-only mirror

BLAS implementation for Intel FPGA

Machine Learning Systems

Vim-fork focused on extensibility and usability

LLM training in simple, raw C/CUDA

Python version of Factored Spectrally Transformed Linear Mixed Models

Genome-wide Efficient Mixed Model Association

Large-scale Bayesian variable selection for R and MATLAB.

Permutation testing for GWAS using FPGAs

Scalable Spike-and-Slab

R-package: Bayesian variable selection, model choice, and regularized estimation for (spatial) generalized additive mixed regression models via stochastic search variable selection with spike-and-s…

You like pytorch? You like micrograd? You love tinygrad! ❤️