Assorted Projects
A popular testbed for AI agents which implements miniaturized versions of several Atari 2600 games.
An experimental falling-sand simulator implemented entirely in JAX with every pixel simulated in parallel on GPU.
An experimental artificial life sim where each cell is controlled by a neural network and weights diffuse with organic matter.
A basic implementation of AlphaZero style learning applied to randomized mazes using the MCTX framework for Monte Carlo Tree Search in JAX.
An implementation of Dreamer V2 written in JAX and applied to MinAtar, with support for running multiple random seeds of an experiment on a single GPU.
An old project where I trained a DQN agent, implemented in theano, to play the board game Hex using self play on top of supervised initialization.