|
..
|
|
_static
|
|
JAX_Vision_transformer.ipynb
|
|
JAX_basic_text_classification.ipynb
|
|
JAX_examples_image_segmentation.ipynb
|
|
JAX_for_LLM_pretraining.ipynb
|
|
JAX_for_PyTorch_users.ipynb
|
|
JAX_image_captioning.ipynb
|
|
JAX_machine_translation.ipynb
|
|
JAX_porting_PyTorch_model.ipynb
|
|
JAX_time_series_classification.ipynb
|
|
JAX_transformer_text_classification.ipynb
|
|
JAX_visualizing_models_metrics.ipynb
|
|
data_loaders_on_cpu_with_jax.ipynb
|
|
data_loaders_on_gpu_with_jax.ipynb
|
|
digits_diffusion_model.ipynb
|
|
digits_vae.ipynb
|
|
getting_started_with_jax_for_AI.ipynb
|
|
JAX_Vision_transformer.md
|
|
JAX_basic_text_classification.md
|
|
JAX_examples_image_segmentation.md
|
|
JAX_for_LLM_pretraining.md
|
|
JAX_for_PyTorch_users.md
|
|
JAX_image_captioning.md
|
|
JAX_machine_translation.md
|
|
JAX_porting_PyTorch_model.md
|
|
JAX_time_series_classification.md
|
|
JAX_transformer_text_classification.md
|
|
JAX_visualizing_models_metrics.md
|
|
conf.py
|
|
contributing.md
|
|
data_loaders.md
|
|
data_loaders_on_cpu_with_jax.md
|
|
data_loaders_on_gpu_with_jax.md
|
|
digits_diffusion_model.md
|
|
digits_vae.md
|
|
examples.md
|
|
getting_started_with_jax_for_AI.md
|
|
index.html
|
|
index.rst
|
|
install.md
|
|
pytorch_users.md
|
|
tutorials.md
|