How to load datasets in JAX with TensorFlow

General Discussion
#1
2 Likes
#3

This is interesting!
Thanks for sharing!

#5

Thanks, glad you found it useful @lgusm