Welcome to JAX DataLoader's documentation! — JAX DataLoader documentation

# Welcome to JAX DataLoader's documentation!

JAX DataLoader is a high-performance data loading library for JAX that provides efficient data loading, preprocessing, and augmentation capabilities.

## Installation

    pip install jax-dataloaders

## Quick Start

    from jax_dataloader import DataLoader, DataLoaderConfig

    # Create a configuration
    config = DataLoaderConfig(
        batch_size=32,
        shuffle=True,
        num_workers=4
    )

    # Create a data loader
    loader = DataLoader(config)

    # Iterate over batches
    for batch in loader:
        # Process your batch
        pass

## Features

- Efficient data loading with JAX

- Support for various data formats (CSV, JSON, Images)

- Data augmentation and preprocessing

- Progress tracking and monitoring

- Memory management and caching

- Multi-GPU support

- Memory monitoring and optimization

## Contents

Contents:

- Installation Guide
  - Requirements
  - Installation Methods
  - Verifying Installation
  - Troubleshooting
- Usage Guide
  - Basic Usage
  - Advanced Usage
  - Best Practices
- API Reference
  - Core Classes
  - Data Loaders
  - Memory Management
  - Progress Tracking
  - Data Augmentation
  - Exceptions
  - Utility Functions
- Examples
  - Basic Examples
  - Loading from Files
  - Advanced Examples
- Tutorials
  - Getting Started with Image Classification
  - Large-Scale Data Processing
  - Multi-GPU Training
  - Custom Data Augmentation
  - Advanced Caching Strategies
- Changelog
  - [Unreleased]
  - [0.1.7] - 2025-03-30
  - [0.1.6] - 2025-03-30
  - [0.1.5] - 2025-03-30
  - [0.1.4] - 2025-03-30
  - [0.1.3] - 2025-03-30
  - [0.1.2] - 2025-03-30
  - [0.1.1] - 2025-03-30
  - [0.1.0] - 2025-03-30

# Indices and tables

- Index

- Module Index

- Search Page