Python codes for results in Tatsuya Haga, Tomoki Fukai, "Extended temporal association memory by modulations of inhibitory circuits".

Analysis of RNNs with different architectures trained on different tasks

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

Research code accompanying AlphaGenome

A Knowledge-grounded framework for Autonomous ML/AI Program Synthesis and Optimization

A curated collection of neuroscience tasks with a common interface.

Monty is a sensorimotor learning framework based on the thousand brains theory of the neocortex.

MultiCell: geometric learning in multicellular development

Deep Imputation for Skeleton data

a mini 2x2 systolic array and PE demo

Open-source release accompanying Gao et al. 2025

Incredibly fast JavaScript runtime, bundler, test runner, and package manager – all in one

Code for the paper Ghosh & Goodman, 2025: Partial recurrence enables robust and efficient computation

PushWorld: A benchmark for manipulation planning with tools and movable obstacles

Quick illustration of how one can easily read books together with LLMs. It's great and I highly recommend it.

MicroK8s is a small, fast, single-package Kubernetes for datacenters and the edge.

Solve puzzles. Learn CUDA.

Code and Data for WMH project

repo for paper https://arxiv.org/abs/2504.13837

[NeurIPS'25] Generalizable Reasoning through Compositional Energy Minimization

cmd-k converts written instruction into executable shell commands, cursor command k that works from the terminal.

A simple, performant and scalable JAX-based world modeling codebase.

Nano vLLM

A machine learning compiler for GPUs, CPUs, and ML accelerators