flukeskywalker

llamafile

Distribute and run LLMs with a single file.

terminalizer

🦄 Record your terminal and generate animated gif images or share a web player

ml-engineering

Machine Learning Engineering Open Book

beartype

Unbearably fast near-real-time hybrid runtime-static type-checking in pure Python.

datatrove

Freeing data processing from scripting madness by providing a set of platform-agnostic customizable pipeline processing blocks.

penzai

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

lets-plot

Multiplatform plotting library based on the Grammar of Graphics

ThunderKittens

Tile primitives for speedy kernels

maxtext

A simple, performant and scalable Jax LLM!

flash-linear-attention

Efficient implementations of state-of-the-art linear attention models in Pytorch and Triton

cccl

CUDA Core Compute Libraries

resource-stream

CUDA related news and material links

pytorch_memlab

Profiling and inspecting memory in pytorch

tueplots

Figure sizes, font sizes, fonts, and more configurations at minimal overhead. Fix your journal papers, conference proceedings, and other scientific publications.

ema-pytorch

A simple way to keep track of an Exponential Moving Average (EMA) version of your pytorch model

open_lm

A repository for research on medium sized language models.

nnsight

The nnsight package enables interpreting and manipulating the internals of deep learned models.

mirage

A multi-level tensor algebra superoptimizer

coldtype

cross-platform display typography in python

bayesian-flow-networks

This is the official code release for Bayesian Flow Networks.

quarto-revealjs-clean

A minimalist and elegant presentation theme for Quarto Reveal.js

torch2jax

Run PyTorch in JAX. 🤝

nanodo

accelerated-scan

Accelerated First Order Parallel Associative Scan

scalax

A simple library for scaling up JAX programs

pypdfplot

Pypdfplot uses pyplot to produce plots in PDF-format with the python script and data files embedded. The plot can be edited by changing filename from .pdf to .py and running the script again.

gptcore

Fast modular code to create and train cutting edge LLMs

optimi

Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers

karras-power-ema-tutorial

statedict2pytree