Slide decks, coding exercises, and quick references for learning the JAX AI Stack

Some CUDA example code with READMEs.

LLM inference in C/C++

LLM101n: Let's build a Storyteller

CUDA Templates and Python DSLs for High-Performance Linear Algebra

Deep Learning model for protein and ligand complex structure prediction from sequences and SMILES

Vulkan/CUDA/HIP/OpenCL/Level Zero/Metal Fast Fourier Transform library

Samples demonstrating how to use the Compute Sanitizer Tools and Public API

🔥 Machine Learning Notebooks

C++ Tip Of The Week

fccf: A command-line tool that quickly searches through C/C++ source code in a directory based on a search string and prints relevant code snippets that match the query.

My CV

A template repo for Python packages

Python programs, usually short, of considerable difficulty, to perfect particular skills.

Curated coding interview preparation materials for busy software engineers

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

RLMeta is a light-weight flexible framework for Distributed Reinforcement Learning Research.

Differentiable, Hardware Accelerated, Molecular Dynamics

Rainbow: Combining Improvements in Deep Reinforcement Learning

A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.

Trainable, memory-efficient, and GPU-friendly PyTorch reproduction of AlphaFold 2

Library for Model Based RL

Graphormer is a general-purpose deep learning backbone for molecular modeling.

Successful Job Applications and Grants

Python code for "Probabilistic Machine learning" book by Kevin Murphy

Trax — Deep Learning with Clear Code and Speed

High-throughput simulation platform for Artificial Intelligence reseach

WESTPA: The Weighted Ensemble Simulation Toolkit with Parallelization and Analysis

Open source code for AlphaFold 2.