🚀✨ A nanoGPT implementation in pure JAX and some infrastructure you can use as template for your own small scale LLM research projects 🚀✨

Machine Learning Systems

Laplace approximations for Deep Learning.

PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)

🙃 A delightful community-driven (with 2,400+ contributors) framework for managing your zsh configuration. Includes 300+ optional plugins (rails, git, macOS, hub, docker, homebrew, node, php, python…

An Open Source software for 3D sound propagation modelling

Tokamax: A GPU and TPU kernel library.

Multiple dispatch

Modular Multi-Agent System for Scientific Research Assistance

Trio – a friendly Python library for async concurrency and I/O

Astropy package for source detection and photometry. Maintainer: @larrybradley

A collection of lightweight interpretability scripts to understand how LLMs think

Simple creation of data classes from dictionaries.

Deep neural network kernel for Gaussian process

Ever used asyncio and wished you hadn't? A tiny (~400 lines) event loop for Python.

Repository with draft content for the Pydantic tutorial at SciPy 2023

A Snakemake workflow for Chandra data reduction

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

Rich is a Python library for rich text and beautiful formatting in the terminal.

Hierarchical Reasoning Model Official Release

A lightweight, local-first, and 🆓 experiment tracking library from Hugging Face 🤗

This repository contains a curated collection of 300+ case studies from over 80 companies, detailing practical applications and insights into machine learning (ML) system design. The contents are o…

Transformer Explained Visually: Learn How LLM Transformer Models Work with Interactive Visualization

Add a tqdm progress bar to your JAX scans and loops.

Python implementation of a method to convert any image into a voronoi tessellation based mosaic. No AI needed!

A library for Jolideco GMM patch prior models