Awesome Open Source
Search
Programming Languages
Languages
All Categories
Categories
About

The Top 23 Flax Open Source Projects

Open source projects categorized as Flax
Categories > Flax
Transformers ⭐ 127,491
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
dependent packages 2,484total releases 125most recent commit 12 days agopypi transformers} Downloads
Pyprobml ⭐ 6,066
Python code for "Probabilistic Machine learning" book by Kevin Murphy
most recent commit 6 months ago
Flaxengine ⭐ 5,246
Flax Engine – multi-platform 3D game engine
most recent commit 5 months ago
Easylm ⭐ 2,087
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
most recent commit 6 months ago
Uvadlc_notebooks ⭐ 1,860
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2022/Spring 2022
most recent commit 6 months ago
Fortuna ⭐ 824
A Library for Uncertainty Quantification.
most recent commit 5 months ago
Score_sde ⭐ 782
Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
most recent commit 2 years ago
Long Range Arena ⭐ 635
Long Range Arena for Benchmarking Efficient Transformers
most recent commit 6 months ago
Jaxrl ⭐ 585
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
most recent commit 2 years ago
Prompt Tuning ⭐ 580
Original Implementation of Prompt Tuning from Lester, et al, 2021
most recent commit 7 months ago
Get Started With Jax ⭐ 487
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.
most recent commit 7 months ago
Tevatron ⭐ 334
Tevatron - A flexible toolkit for neural retrieval research and development.
dependent packages 1total releases 5most recent commit 6 months agopypi tevatron} Downloads
Commonlooputils ⭐ 261
CLU lets you write beautiful training loops in JAX.
dependent packages 11total releases 13most recent commit 6 months agopypi clu} Downloads
Gradcache ⭐ 254
Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint
total releases 1most recent commit 6 months ago
Flaxmodels ⭐ 201
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
dependent packages 2total releases 4most recent commit 10 months agopypi flaxmodels} Downloads
Orbax ⭐ 187
Orbax provides common utility libraries for JAX users.
dependent packages 9total releases 28most recent commit 5 months agopypi orbax-checkpoint} Downloads
Jaxseq ⭐ 172
Train very large language models in Jax.
most recent commit 8 months ago
Flaxapi ⭐ 129
Old repository with C# Editor and C# API for creating games in Flax Engine
most recent commit 4 years ago
Flax ⭐ 123
general purpose programming language, in the vein of C++
most recent commit 3 years ago
Efficientnet Jax ⭐ 121
EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
most recent commit 6 months ago
Koclip ⭐ 117
KoCLIP: Korean port of OpenAI CLIP, in Flax
most recent commit 10 months ago
Flaxdocs ⭐ 83
Flax Engine Documentation
most recent commit 6 months ago
Easydel ⭐ 83
EasyDeL is an OpenSource Library to make your training faster and more Optimized With cool Options for training and serving Both in Python And Mojo🔥
dependent packages 1total releases 38most recent commit 5 months agopypi EasyDeL} Downloads
The 7 Most Depended On Flax Open Source Projects
Transformers ⭐ 127,491
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
dependent packages 2,484total releases 125latest release November 15, 2023most recent commit 12 days agopypi transformers} Downloads
Commonlooputils ⭐ 261
CLU lets you write beautiful training loops in JAX.
dependent packages 11total releases 13latest release October 02, 2023most recent commit 6 months agopypi clu} Downloads
Orbax ⭐ 187
Orbax provides common utility libraries for JAX users.
dependent packages 9total releases 28latest release December 07, 2023most recent commit 5 months agopypi orbax-checkpoint} Downloads
Flaxmodels ⭐ 201
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
dependent packages 2total releases 4latest release August 11, 2023most recent commit 10 months agopypi flaxmodels} Downloads
Jax Resnet ⭐ 67
Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
dependent packages 1total releases 2latest release May 25, 2021most recent commit 2 years agopypi jax-resnet} Downloads
Tevatron ⭐ 334
Tevatron - A flexible toolkit for neural retrieval research and development.
dependent packages 1total releases 5latest release February 22, 2022most recent commit 6 months agopypi tevatron} Downloads
The 10 Latest Releases In Flax Open Source Projects
Orbax ⭐ 187
Orbax provides common utility libraries for JAX users.
dependent packages 9total releases 28latest release December 07, 2023most recent commit 5 months agopypi orbax-checkpoint} Downloads
Transformers ⭐ 127,491
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
dependent packages 2,484total releases 125latest release November 15, 2023most recent commit 12 days agopypi transformers} Downloads
Redco ⭐ 35
MLSys Workshop NeurIPS 2023 - Redco: A Lightweight Tool to Automate Distributed Training and Inference
total releases 11latest release November 08, 2023most recent commit 6 months ago
Ramsey ⭐ 10
Probabilistic deep learning using JAX
total releases 5latest release October 23, 2023most recent commit 8 months ago
Commonlooputils ⭐ 261
CLU lets you write beautiful training loops in JAX.
dependent packages 11total releases 13latest release October 02, 2023most recent commit 6 months agopypi clu} Downloads
Flaxmodels ⭐ 201
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
dependent packages 2total releases 4latest release August 11, 2023most recent commit 10 months agopypi flaxmodels} Downloads
Helx ⭐ 9
Interoperating between (Deep) Reiforcement Learning libraries
total releases 5latest release January 31, 2023most recent commit 5 months ago
Flaim ⭐ 12
Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.
total releases 5latest release January 27, 2023most recent commit a year ago
Safejax ⭐ 13
Serialize JAX, Flax, Haiku, or Objax model params with `safetensors`
total releases 6latest release January 05, 2023most recent commit a year ago
Categories
Privacy | About | Terms | Follow Us On Twitter

Downloads, Dependent Repos, Dependent Packages, Total Releases, Latest Releases data powered by Libraries.io.

Copyright 2018-2024 Awesome Open Source.  All rights reserved.