nsys-jax pure-Python CI

Add an option to test-pax.sh to enable XLA cuDNN flash attention #249

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
notebook	1m 15s
ruff	5s
mypy	1m 1s
combine	1m 44s
	4m 5s