jax
4f935636 - [pallas] Support for setting explicit backends to pallas_call.

Commit

1 year ago

[pallas] Support for setting explicit backends to pallas_call. PiperOrigin-RevId: 688511303

Author

cperivol

Committer

Google-ML-Automation

Parents