jax
7305b64f - Add cuda to _nonexperimental_plugins. It passed existing GPU tests, and have set up continuous testing job.

Commit

2 years ago

Add cuda to _nonexperimental_plugins. It passed existing GPU tests, and have set up continuous testing job. PiperOrigin-RevId: 590328635

References

Author

Jieying Luo

Committer

a-googler

Parents