Skip to main content

Ctrl+K

Grain

Get started

Installing Grain
Choice of API

Data sources

Protocol
Bagz
ArrayRecord
Parquet
TFDS
HuggingFace
PyTorch
GCS
Amazon S3

Dataset

Basics
Advanced usage
Transformations
Performance debugging

DataLoader

Samplers
Transformations
DataLoader tutorial

Tutorials

JAX training tips

API reference

grain package
Change log

For contributors

Behind the Scenes
Contributing to Grain

Repository
Open issue

.rst

grain.sharding module

Contents

List of Members

`grain.sharding` module#

List of Members#

previous

grain.samplers module

next

grain.sources module

Contents

List of Members

By Grain team

© Copyright 2024, Grain team.