PR GPU tests

Add fully configurable activation checkpointing #4490

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
gpu-2.1.0 / pytest-gpu	4m 24s
gpu-2.1.0-flash2 / pytest-gpu	4m 7s
	8m 31s