Slides of my talks and lectures

✨ Magical shell history

♟️ Vectorized RL game environments in JAX

Animation engine for explanatory math videos

Open source code for EigenGame.

RL Environments in JAX 🌍

Evolution Strategies in JAX 🦎

Repository for the main Dockerfile with the OpenWorm software stack and project-wide issues

🦀 Small exercises to get you used to reading and writing Rust code!

Tile primitives for speedy kernels

Make Zotero effective for us LaTeX holdouts

Kolmogorov Arnold Networks

Puzzles for exploring transformers

Pen and paper exercises in machine learning

Complete collection of my PGF/TikZ figures.

A minimal GPU design in Verilog to learn how GPUs work from the ground up

[ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.

Recipes for using Python's pandas library

2D vector & raster editor that melds traditional layers & tools with a modern node-based, non-destructive, procedural workflow.

Macro to draw architecture of convolutional neural nets in tikz.

Code for the book "The Elements of Differentiable Programming".

Spectral Neural Operator

Optax is a gradient processing and optimization library for JAX.

Understanding Deep Learning - Simon J.D. Prince

18.335 - Introduction to Numerical Methods course

Pseudospectral Kolmogorov Flow Solver

A conda-forge distribution.