xla
d780e55c - Add jax_device context manager to control the device target (#9382)

Commit

200 days ago

Add jax_device context manager to control the device target (#9382)

References

#9382 - Add jax_device context manager to control the device target

Author

zzzwen

zzzwen

Parents

Loading