JAX Framework# Getting started with JAX and DALI Prerequisites Quick start Running DALI with JAX Batching GPU acceleration Hardware accelerated decoding Random shuffling Augmentations Multiple GPUs Technical details Where to go next Training a neural network with DALI and JAX Training with multiple GPUs Training with automatic parallelization Training with pmapped iterator Training neural network with DALI and Flax Multiple GPUs with DALI and FLAX Training neural network with DALI and Paxml Getting started with T5X and DALI Prerequisites Peekable iterator for T5X Where to go next
