Closed SamDuffield closed 2 months ago
Passing the random_key rather than seed is better for JAX composability
Passing the random_key rather than seed is better for JAX composability