jax
ed9a4c29 - add `jax.threefry_partitionable` context manager

Commit

2 years ago

add `jax.threefry_partitionable` context manager

References

#18338 - add `jax.threefry_partitionable` context manager

Author

froystig

froystig

Committer

froystig

froystig

Parents

Loading