SIMD Acceleration
19 data augmentation transforms with AVX2/NEON SIMD acceleration for maximum throughput.
Includes AutoAugment policies for ImageNet, CIFAR10, and SVHN.
Lock-Free Architecture
Custom SPSC (Single Producer Single Consumer) queues with zero-copy design.
No mutex contention for maximum parallelism.
Multi-Format Support
Images (JPEG, PNG, WebP), video (H.264, HEVC), tabular data (CSV, Parquet).
WebDataset TAR format with remote storage (HTTP, S3, GCS).
GPU Integration
Direct PyTorch/TensorFlow/JAX tensor conversion. Optional GPU-accelerated
JPEG decoding with nvJPEG for ultra-low latency.