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

Code examples and resources for DBRX, a large language model developed by Databricks

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.

Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators

BLAS implementation for Intel FPGA

Introduction to 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! ❤️

LAPACK development repository