flax
2e7012da - removed slashes from improved _prefetch assert.

Commit

5 years ago

removed slashes from improved _prefetch assert.

References

#346 - Added an error message to the assert in jax_utils.prefetch_to_device(*)

Author

levskaya

levskaya

Parents

Loading