GitHub repository: mikiec84/pt-jax
Description: Path Tracing in JAX