The n-gram Language Model

MLX: An array framework for Apple silicon

Schedule-Free Optimization in PyTorch

seqax = sequence modeling + JAX

This is our submission to the Event-based eye Tracking challenge part of the CVPR 2024 Workshop AI for Streaming.

Deep Hierarchical Planning from Pixels

SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.

Optax is a gradient processing and optimization library for JAX.

Animation engine for explanatory math videos

Implementation of the experiments in the paper Sleep Deprivation in the Forward-Forward algorithm.

The Mojo Programming Language

StableLM: Stability AI Language Models

An interpreter for RASP as described in the ICML 2021 paper "Thinking Like Transformers"

Predictive coding in Python

Locally run an Instruction-Tuned Chat-Style LLM

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

An implementation of Deep Q-Learning playing Connect Four.

Precision agriculture image processing suite for UAS-acquired multipsectral imagery on GCP

Training neural models with structured signals.

An introduction to ARM64 assembly on Apple Silicon Macs

Reproducing CIFAR10 experiment results from the original ResNet paper using Pytorch in IPython notebook.

PyTorch Dual-Attention LSTM-Autoencoder For Multivariate Time Series

A comprehensive list of awesome contrastive self-supervised learning papers.

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

MuZero

Implementation of Reinforcement Learning Algorithms. Python, OpenAI Gym, Tensorflow. Exercises and Solutions to accompany Sutton's Book and David Silver's course.

Python Implementation of Reinforcement Learning: An Introduction

A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API

Learning trajectory representations using self-supervision and programmatic supervision.

Vim - the text editor - for macOS