Discover the Best Open-Source Flax Projects Sorted by Stars
Get Insights into Stars Growth, Contributions, Diversity, Bus Factor, and Community Governance. Optimize Your Project's Performance and Collaborate with a Thriving Community.
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Python code for "Probabilistic Machine learning" book by Kevin Murphy
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2022/Spring 2022
Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
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.
Get updates on the fastest growing repos and cool stats about GitHub right in your inbox
Once per month. No spam.