magehrig

torchtitan

A native PyTorch Library for large model training

AI-scripts

Some handy AI scripts

lazygit

simple terminal UI for git commands

ThunderKittens

Tile primitives for speedy kernels

penzai

A JAX research toolkit for building, editing, and visualizing neural networks.

llm.c

LLM training in simple, raw C/CUDA

dora

DORA (Dataflow-Oriented Robotic Architecture) is middleware designed to streamline and simplify the creation of AI-based robotic applications. It offers low latency, composable, and distributed dataflow capabilities. Applications are modeled as directed graphs, also referred to as pipelines.

EvDeblurNeRF

Code accompanying the CVPR24 paper "Mitigating Motion Blur in Neural Radiance Fields with Events and Frames"

ml-engineering

Machine Learning Engineering Open Book

transformer-debugger

awesome-mojo

A curated list of awesome Mojo 🔥 frameworks, libraries, software and resources

LEOD

Code release for CVPR 2024 paper LEOD: Label-Efficient Object Detection for Event Cameras

mamba.py

A simple and efficient Mamba implementation in pure PyTorch and MLX.

vim

The official Vim repository

mamba-minimal

Simple, minimal implementation of the Mamba SSM in one file of PyTorch.

PowerInfer

High-speed Large Language Model Serving on PCs with Consumer-grade GPUs

l4casadi

Use PyTorch Models with CasADi for data-driven optimization or learning-based optimal control. Supports Acados.

mypy

Optional static typing for Python

equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

mamba

Mamba SSM architecture

gpt-fast

Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.

mip-splatting

[CVPR'24 Best Student Paper] Mip-Splatting: Alias-free 3D Gaussian Splatting

gaussian-splatting

Original reference implementation of "3D Gaussian Splatting for Real-Time Radiance Field Rendering"

tinyraytracer

A brief computer graphics / rendering course

S5

llama2.mojo

Inference Llama 2 in one file of pure 🔥

HMNet_pth

PyTorch implementation of Hierarchical Neural Memory Network

m3ed

M3ED Dataset

dynamic-sparse-flash-attention

mojo

The Mojo Programming Language