🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFl...
Python code for "Probabilistic Machine learning" book by Kevin Murphy
Flax Engine – multi-platform 3D game engine
Repository of Jupyter notebook tutorials for teaching the Deep Learning ...
Official code for Score-Based Generative Modeling through Stochastic Dif...
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, Fla...
Tevatron - A flexible toolkit for neural retrieval research and developm...
Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Con...
CLU lets you write beautiful training loops in JAX.
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResN...
Train very large language models in Jax.
general purpose programming language, in the vein of C++
Unofficial JAX implementations of deep learning research papers