#

jax-dataloader

Here are 2 public repositories matching this topic...

BirkhoffG / jax-dataloader

Pytorch-like dataloaders in JAX.

deep-learning tensorflow pytorch dataset dataloader datasets jax huggingface-datasets jax-dataloader

Updated Oct 18, 2024
Jupyter Notebook

bsc-quantic / tn4ml

Tensor Networks for Machine Learning

machine-learning flax jax quimb optax tensornetworks jax-dataloader

Updated Jan 8, 2025
Python

Improve this page

Add a description, image, and links to the jax-dataloader topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the jax-dataloader topic, visit your repo's landing page and select "manage topics."