Python flax

Open-source Python projects categorized as flax
Edit details
Topics: Jax Deep Learning Machine Learning NLP Computer Vision
Power Real-Time Data Analytics at Scale
Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.
www.influxdata.com
Sponsored

Top 15 Python flax Projects

  • transformers

    🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

    • Project mention: Maxtext: A simple, performant and scalable Jax LLM | news.ycombinator.com | 2024-04-23

    Is t5x an encoder/decoder architecture?

    Some more general options.

    The Flax ecosystem

    https://github.com/google/flax?tab=readme-ov-file

    or dm-haiku

    https://github.com/google-deepmind/dm-haiku

    were some of the best developed communities in the Jax AI field

    Perhaps the “trax” repo? https://github.com/google/trax

    Some HF examples https://github.com/huggingface/transformers/tree/main/exampl...

    Sadly it seems much of the work is proprietary these days, but one example could be Grok-1, if you customize the details. https://github.com/xai-org/grok-1/blob/main/run.py

  • EasyLM

    Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.

    • Project mention: Maxtext: A simple, performant and scalable Jax LLM | news.ycombinator.com | 2024-04-23

  • InfluxDB

    Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

    InfluxDB logo

  • fortuna

    A Library for Uncertainty Quantification.

  • long-range-arena

    Long Range Arena for Benchmarking Efficient Transformers

    • Project mention: The Secret Sauce behind 100K context window in LLMs: all tricks in one place | news.ycombinator.com | 2023-06-17

    https://github.com/google-research/long-range-arena

  • prompt-tuning

    Original Implementation of Prompt Tuning from Lester, et al, 2021

  • GradCache

    Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint

  • flaxmodels

    Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.

  • WorkOS

    The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.

    WorkOS logo

  • EasyDeL

    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🔥

    • Project mention: FLaNK Stack 26 February 2024 | dev.to | 2024-02-26

  • jax-models

    Unofficial JAX implementations of deep learning research papers

  • jax-resnet

    Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).

  • omega

    A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environments. (by hr0nix)

  • flaxOptimizers

    A collection of optimizers, some arcane others well known, for Flax.

  • flaim

    Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.

  • HJxB

    Continuous-Time/State/Action Fitted Value Iteration via Hamilton-Jacobi-Bellman (HJB)

  • h-former

    H-Former is a VAE for generating in-between fonts (or combining fonts). Its encoder uses a Point net and transformer to compute a code vector of glyph. Its decoder is composed of multiple independent decoders which act on a code vector to reconstruct a point cloud representing a glpyh.

  • SaaSHub

    SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives

    SaaSHub logo
NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020).

Python flax related posts

Index

What are some of the best open-source flax projects in Python? This list will help you:

Project Stars
1 transformers 124,557
2 EasyLM 2,228
3 fortuna 850
4 long-range-arena 680
5 prompt-tuning 620
6 GradCache 305
7 flaxmodels 222
8 EasyDeL 148
9 jax-models 138
10 jax-resnet 98
11 omega 36
12 flaxOptimizers 28
13 flaim 14
14 HJxB 11
15 h-former 5

Sponsored
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives
www.saashub.com