Jax implementation of "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"
? Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
? Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.
StarCraft II Learning Environment
Python code for "Probabilistic Machine learning" book by Kevin Murphy
Learn Deep Reinforcement Learning in 60 days! Lectures & Code in Python. Reinforcement Learning + Deep Learning
DataSphereStudio is a one stop data application development& management portal, covering scenarios including data exchange, desensitization/cleansing, analysis/mining, quality measurement, visualization, and task scheduling.
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.
Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
From the Tensor to Stable Diffusion, a rough outline for a 1 week course.