flax
bec1c80d - Merge pull request #3308 from anuragarnab:patch-3

Commit

2 years ago

Merge pull request #3308 from anuragarnab:patch-3 PiperOrigin-RevId: 563345395

References

#3308 - Minor fix to jax_utils.prefect_to_device

Author

a-googler

a-googler

Parents

Loading