jax
d7d0aa94 - Move PRNG GPU lowering from jaxlib into JAX.

Commit

1 year ago

Move PRNG GPU lowering from jaxlib into JAX. PiperOrigin-RevId: 738398099

References

#27212 - Move PRNG GPU lowering from jaxlib into JAX.

Author

dfm

dfm

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading