jax
ad4c9ab8 - [Mosaic GPU] Avoid failing when importing profiler.py even if lib.mosaic_gpu is unavailable.

Commit

1 year ago

[Mosaic GPU] Avoid failing when importing profiler.py even if lib.mosaic_gpu is unavailable. PiperOrigin-RevId: 647626620

References

Author

Google-ML-Automation

Committer

a-googler

Parents