nsys-jax pure-Python CI

Fixes Imagen sampling example and updates container #133

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
mypy	52s
combine	1m 11s
notebook	1m 18s
ruff	6s
	3m 27s