Data module (AtomicData, Batch, readers/writers)

Core classes

AtomicData	Atomic data structure for molecular systems.
Batch	Graph-aware batch built on MultiLevelStorage.

I/O and pipelines

AtomicDataZarrWriter	Writer for serializing AtomicData into Zarr stores.
AtomicDataZarrReader	Reader for loading AtomicData from Zarr stores.
Dataset	AtomicData-native dataset that bypasses TensorDict conversion.
DataLoader	Batch-iterating data loader that yields Batch.
Reader	Abstract base class for data readers.

Write configuration

ZarrArrayConfig	Configuration for Zarr array compression, chunking, and sharding.
ZarrWriteConfig	Top-level write configuration for AtomicDataZarrWriter.