Index A | B | D | E | F | J | M | P | R | S A as_jit_compatible() (jaxio.Dataset method) B batch() (jaxio.Dataset method) D Dataset (class in jaxio) E enumerate() (jaxio.Dataset method) F filter() (jaxio.Dataset method) fmap() (jaxio.Dataset method) from_next_fn() (jaxio.Dataset class method) from_pytree_slices() (jaxio.Dataset class method) J jit() (jaxio.Dataset method) M map() (jaxio.Dataset method) P prefetch() (jaxio.Dataset method) R repeat() (jaxio.Dataset method) S sleep() (jaxio.Dataset method)