JAX

• JAX Plugin API reference
  • DALIGenericIterator
  • data_iterator()
• JAX Framework
  • Getting started with JAX and DALI
  • Training a neural network with DALI and JAX
  • Training with multiple GPUs
  • 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
• Running JAX in DALI pipeline