flax
Tracked open-source repos tagged flax, sorted by stars.
Related topics
Topics that frequently appear alongside flax on the same repo.
Recent risers
Repos created in the last 90 days, tagged flax.
No new repos tagged with this topic in the last 90 days.
- #1★ 7,131+0Star change over the last 7 days
- #2
Flax Engine – multi-platform 3D game engine
★ 6,988+5Star change over the last 7 days - #3
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
★ 3,190+5Star change over the last 7 days - #4
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
★ 2,515+1Star change over the last 7 days - #5
Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/Gym, NVIDIA Isaac Lab, MuJoCo Playground and other environments
★ 1,095+2Star change over the last 7 days - #6
Tevatron - Unified Document Retrieval Toolkit across Scale, Language, and Modality. Demo in SIGIR 2023, SIGIR 2025.
★ 749+0Star change over the last 7 days - #7★ 534-1Star change over the last 7 days
