Skip to main content
Back to top
Ctrl
+
K
Jax AI Stack
Installing the stack
Tutorials
Getting started with JAX for AI
Variational autoencoder (VAE) and debugging in JAX
Train a diffusion model for image generation with JAX for AI
Visualize JAX model metrics with TensorBoard
Introduction to Data Loaders
Introduction to Data Loaders on CPU with JAX
Introduction to Data Loaders on GPU with JAX
From PyTorch to JAX
JAX for PyTorch users
Porting a PyTorch model to JAX
Example applications
Train a miniGPT language model with JAX
Basic text classification with 1D CNN
Text classification with a transformer language model using JAX
Machine Translation with encoder-decoder transformer model
Image segmentation with UNETR model
Image Captioning with Vision Transformer (ViT) model
Train a Vision Transformer (ViT) for image classification with JAX
Time series classification with CNN
Contribute to documentation
Search
Error
Please activate JavaScript to enable the search functionality.
Ctrl
+
K
