#Flax
Showing 15 of 15 repositories tagged #flax, ranked by stars
Python code for "Probabilistic Machine learning" book by Kevin Murphy
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
A Library for Uncertainty Quantification.
Long Range Arena for Benchmarking Efficient Transformers
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.
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
Original Implementation of Prompt Tuning from Lester, et al, 2021
Orbax provides common checkpointing and persistence utilities for JAX users
Unofficial JAX implementations of deep learning research papers
A JAX-based deep learning framework for image segmentation using diffusion models.
JAX implementations of core Deep RL algorithms
Amos optimizer with JEstimator lib.
A simple and intuitive 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.
Generative neural networks for 3D terrain.