#计算机科学#🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.
#计算机科学#Python code for "Probabilistic Machine learning" book by Kevin Murphy
#游戏引擎# Flax Engine - 多平台 3D 游戏引擎
#计算机科学#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.
#计算机科学#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 t...
#计算机科学#JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
#自然语言处理#Original Implementation of Prompt Tuning from Lester, et al, 2021
Tevatron - Unified Document Retrieval Toolkit across Scale, Language, and Modality. Demo in SIGIR 2023, SIGIR 2025.
#计算机科学#Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint
Orbax provides common checkpointing and persistence utilities for JAX users
#自然语言处理#A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
Accelerate, Optimize performance with streamlined training and serving options with JAX.
#计算机科学#Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
#计算机科学#Train very large language models in Jax.