API Reference
Experimental API
Called at the beginning of an epoch. Must be jittable.
list[Any]
list
Any