flax

#自然语言处理#为 Jax、PyTorch 和 TensorFlow 打造的先进的自然语言处理

#计算机科学#🤗 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

翻译 - 用于“机器学习：概率视角”的Python代码（第2版）

#游戏引擎# 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...

#自然语言处理#Original Implementation of Prompt Tuning from Lester, et al, 2021

#计算机科学#JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.

Tevatron - A flexible toolkit for neural retrieval research and development.

#计算机科学#Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint

Orbax provides common checkpointing and persistence utilities for JAX users

CLU lets you write beautiful training loops in JAX.

#自然语言处理#A Jax-based library for designing and training transformer models from scratch.

#计算机科学#Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.

Accelerate, Optimize performance with streamlined training and serving options with JAX.