Tutorial: 3D rendering with ~100 lines of pure Jax code
google-research.github.io