ramsey-devs / ramsey

Probabilistic deep learning using JAX
https://ramsey.rtfd.io
Apache License 2.0
13 stars 3 forks source link

Use TFDS iterators #44

Open dirmeier opened 7 months ago

dirmeier commented 7 months ago

The current data iterators are slow. We could use the tf.data iterators.

Does that work with dynamic input/output dimensions?