wip

  • 17th July 2026

fae: FLUX.1 SAE training on TPUs

jax diffusion wip

An implementation of MMDiT in jax, + SAE training on TPUs

Read more →
  • 30th January 2025

Codeball environment for PufferLib

3d wip

A 3D RL environment

Read more →