configs/default_cifar10_configs.py (1 line):
	- line 18: training.n_jitted_steps = 5  # TODO: important flag!