jax
c626a86f - Added out_sharding to jax.random.generalized_normal sampler

Commit

95 days ago

Added out_sharding to jax.random.generalized_normal sampler Description: - Added out_sharding to jax.random.generalized_normal sampler - Added a test

References

Author

vfdev-5

Committer

vfdev-5

Parents