Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
-
Updated
Mar 16, 2025 - Jupyter Notebook
#
jax
Here are 22 public repositories matching this topic...
Notebooks for the Practicals at the Deep Learning Indaba 2022.
reinforcement-learning transformers bayesian-inference jax graph-neural-networks deep-generative-models
-
Updated
Apr 3, 2024 - Jupyter Notebook
💁 Awesome Treasure of Transformers Models for Natural Language processing contains papers, videos, blogs, official repo along with colab Notebooks. 🛫☑️
python nlp natural-language-processing awesome tensorflow pytorch transformer speech-recognition seq2seq pretrained-models language-models natural-language-generation nlp-library language-model bert natural-language-understanding jax pytorch-transformers model-hub
-
Updated
Aug 1, 2025 - Jupyter Notebook
Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.
python data-science machine-learning natural-language-processing reinforcement-learning computer-vision deep-learning mxnet book notebook tensorflow keras pytorch kaggle hyperparameter-optimization recommender-system gaussian-processes jax
-
Updated
Aug 18, 2024 - Python
Colab notebooks showcasing experiments on MPPI (model predictive path integral control) and CBF (control barrier function). Utilizes jax to accelerate computation.
-
Updated
Feb 25, 2025 - Jupyter Notebook
The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.
-
Updated
Nov 29, 2023 - Jupyter Notebook
Collection of marimo tutorials which encompass notebook/app examples in varying domains - CS/AI/ML
data-science machine-learning computer-vision tensorflow image-processing pytorch artificial-intelligence recommender-system jax llm marimo marimo-notebook
-
Updated
Mar 9, 2025 - Python
JAX Tutorial notebooks : basics, crash & tips, usage of optax/JaxOptim/Numpyro
-
Updated
Feb 25, 2025 - Jupyter Notebook
Physics Informed Neural Networks (PINNs) + SPINNs + HyperPINNs + Adaptative Loss Weights with JAX 📓 Check out our various notebooks to get started
-
Updated
Aug 1, 2025 - Jupyter Notebook
Repository of notes, code and notebooks in Python for the book "Reinforcement Learning: An Introduction" by Richard S. Sutton and Andrew G. Barto
-
Updated
Aug 7, 2025 - Jupyter Notebook
GSoC'22 @ TensorFlow Notebooks, Code and much more
-
Updated
Sep 20, 2022 - Jupyter Notebook
This repository provides a selection of very basic and minimal notebooks for various NLP tasks in written in Jax.
-
Updated
Jan 14, 2021 - Jupyter Notebook
Jupyter notebooks about Programming, Statistics and Math
python optimization dsl continuation monad functor monoid constraint-programming probabilistic-programming tail-call-optimization bayes resampling differentiation coroutine codata jax array-programming corecursion concatenative-programming applicative-functor
-
Updated
Jan 3, 2025 - Jupyter Notebook
A collection of Python notebooks showing various ML techniques.
-
Updated
Aug 1, 2024 - Jupyter Notebook
Notebook and scripts of Prediction, Learning, and Games by Nicolò Cesa-Bianchi
-
Updated
Aug 14, 2023 - Jupyter Notebook
Miscellaneous Jupyter notebooks.
-
Updated
Jun 9, 2025 - Jupyter Notebook
A simple, easy-to-understand library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with simplified mathematical explanations. Made as part of my journey for learning and experimenting with generative AI.
machine-learning deep-learning tensorflow attention image-generation flax unet diffusion edm distributed-training image2image ai-research jax diffusion-models ddpm score-based-generative-modeling ddim stable-diffusion generative-ai karras
-
Updated
May 6, 2025 - Jupyter Notebook
TPU Container Integration for Jupyter Notebook & TensorFlow
python docker jupyter notebook reproducible-research tensorflow container jupyter-notebook oci tpu jax podman
-
Updated
Oct 29, 2023 - Dockerfile
Improve this page
Add a description, image, and links to the jax topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the jax topic, visit your repo's landing page and select "manage topics."