v.0.2.8

Added HMS dataset from kaggle

v.0.2.7

v0.2.7

new version

v.0.2.6

v0.2.6

version bump

Automatic reset

This release fixes the bug in which the dataloader's index does not automatically reset, i.e. this works now as expected:

import jax.numpy as jnp
from jaxonloader import get_mnist
from jaxonloader.dataloader import make


def test_reset_with_jit():
    train, test = get_mnist()

    train_loader, index = make(train, batch_size=64, drop_last=True, jit=True)
    first_sample, _, _ = train_loader(index)

    while it := train_loader(index):
        x, index, done = it
        if done:
            break

    next_first_sample, _, _ = train_loader(index)
    assert jnp.allclose(first_sample, next_first_sample, atol=1e-6)


if __name__ == "__main__":
    test_reset_with_jit()

v0.2.4

Added new docs!

0.2.3

circular imports