SUMMARY

• Dataloader for JAX
• Examples
  • Vision Transformer
• API
  • Core API
  • Dataset
  • Loaders
    • Dataloader
    • JAX Dataloader
    • Pytorch-backed Dataloader Use Pytorch to load batches. It requires pytorch to be installed.
    • Tensorflow-backed Dataloader
  • Utils