jax
4fe9e596 - Add jax[cuda12] install variation for using cuda plugin.

Commit

2 years ago

Add jax[cuda12] install variation for using cuda plugin. PiperOrigin-RevId: 590342149

References

Author

Jieying Luo

Committer

a-googler

Parents