Native data handling for JAX
Project description
Jax Datasets
data
To load data, first subclass the jax_data.Dataset
class, implementing the len and getitem methods.
The jax_data.Dataloader
class is a simplified adaptation of torch.utils.data.DataLoader.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
jax-data-0.1.1.tar.gz
(2.0 kB
view hashes)