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.constants module

Contents

List of Constants

`grain.constants` module#

List of Constants#

previous

grain.checkpoint module

next

grain.experimental module

Contents

List of Constants

By Grain team

© Copyright 2024, Grain team.