Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
Projects in Awesome Lists tagged with flax
A curated list of projects in awesome lists tagged with flax .
https://github.com/huggingface/transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
bert deep-learning flax hacktoberfest jax language-model language-models machine-learning model-hub natural-language-processing nlp nlp-library pretrained-models python pytorch pytorch-transformers seq2seq speech-recognition tensorflow transformer
Last synced: 16 Dec 2024
https://github.com/huggingface/diffusers
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
deep-learning diffusion flax hacktoberfest image-generation image2image jax latent-diffusion-models pytorch score-based-generative-modeling stable-diffusion stable-diffusion-diffusers text2image
Last synced: 16 Dec 2024
https://github.com/probml/pyprobml
Python code for "Probabilistic Machine learning" book by Kevin Murphy
blackjax colab flax jax jupyter-notebooks machine-learning numpyro pml probabilistic-programming pymc3 pyro pytorch tensorflow
Last synced: 17 Dec 2024
https://github.com/flaxengine/flaxengine
Flax Engine – multi-platform 3D game engine
c-sharp cpp engine flax flax-engine flaxengine game-development game-engine gamedev multi-platform
Last synced: 17 Dec 2024
https://github.com/FlaxEngine/FlaxEngine
Flax Engine – multi-platform 3D game engine
c-sharp cpp engine flax flax-engine flaxengine game-development game-engine gamedev multi-platform
Last synced: 28 Oct 2024
https://github.com/phlippe/uvadlc_notebooks
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
deep-learning flax jax optax pytorch pytorch-lightning tutorial tutorials uvadlc
Last synced: 29 Nov 2024
https://github.com/young-geng/easylm
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
chatbot deep-learning flax jax language-model large-language-models llama natural-language-processing transformer
Last synced: 18 Dec 2024
https://github.com/young-geng/EasyLM
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
chatbot deep-learning flax jax language-model large-language-models llama natural-language-processing transformer
Last synced: 25 Oct 2024
https://github.com/yang-song/score_sde
Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
controllable-generation diffusion-models flax generative-models iclr-2021 inverse-problems jax score-based-generative-modeling score-matching stochastic-differential-equations
Last synced: 15 Dec 2024
https://github.com/awslabs/fortuna
A Library for Uncertainty Quantification.
ai bayesian-inference calibration conformal-prediction deep-learning flax jax machine-learning ml model-calibration neural-networks uncertainty uncertainty-calibration uncertainty-estimation uncertainty-quantification
Last synced: 30 Oct 2024
https://github.com/google-research/long-range-arena
Long Range Arena for Benchmarking Efficient Transformers
attention deep-learning flax jax nlp transformers
Last synced: 20 Dec 2024
https://github.com/gordicaleksa/get-started-with-jax
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.
deep-learning flax haiku jax jupyter lax learn-jax machine-learning numpy optax python tutorial xla
Last synced: 16 Dec 2024
https://github.com/google-research/prompt-tuning
Original Implementation of Prompt Tuning from Lester, et al, 2021
flax jax language-model machine-learning nlp prompt-tuning
Last synced: 18 Dec 2024
https://github.com/ikostrikov/jaxrl
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
batch-reinforcement-learning behavioral-cloning continuous-control deep-deterministic-policy-gradient deep-learning deep-reinforcement-learning flax gym jax offline-reinforcement-learning reinforcement-learning sac soft-actor-critic
Last synced: 14 Dec 2024
https://github.com/texttron/tevatron
Tevatron - A flexible toolkit for neural retrieval research and development.
dense-retrieval dpr flax information-retrieval jax pytorch question-answering transformer
Last synced: 11 Nov 2024
https://github.com/google/orbax
Orbax provides common checkpointing and persistence utilities for JAX users
Last synced: 02 Nov 2024
https://github.com/hmunachi/nanodl
A Jax-based library for designing and training transformer models from scratch.
attention attention-mechanism deep-learning distributed-training flax gpt jax llama machine-learning mistral nlp transformer
Last synced: 14 Dec 2024
https://github.com/matthias-wright/flaxmodels
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
deep-learning flax gpt2 jax pretrained-models resnet stylegan2 vgg
Last synced: 14 Dec 2024
https://github.com/erfanzar/easydel
Accelerate, Optimize performance with streamlined training and serving options with JAX.
easydel flax gpt jax nlp optax transformers
Last synced: 21 Dec 2024
https://github.com/erfanzar/EasyDeL
Accelerate, Optimize performance with streamlined training and serving options with JAX.
easydel flax gpt jax machine-learning mojo nlp optax transformers
Last synced: 16 Nov 2024
https://github.com/sea-snell/jaxseq
Train very large language models in Jax.
deep-learning flax gpt2 gpt3 huggingface jax language-models opt
Last synced: 06 Dec 2024
https://github.com/DarshanDeshpande/jax-models
Unofficial JAX implementations of deep learning research papers
artificial-intelligence computer-vision convolutional-neural-networks deep-learning flax jax machine-learning research-paper-implementation transformers
Last synced: 18 Nov 2024
https://github.com/flax-lang/flax
general purpose programming language, in the vein of C++
c-plus-plus compiler flax language language-design llvm programming-language
Last synced: 16 Nov 2024
https://github.com/jaketae/koclip
KoCLIP: Korean port of OpenAI CLIP, in Flax
flax jax openai-clip roberta vision-transformer
Last synced: 30 Nov 2024
https://github.com/rwightman/efficientnet-jax
EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
efficientnet flax flax-linen jax mixnet mobilenetv2 mobilenetv3 objax tpu
Last synced: 05 Dec 2024
https://github.com/yang-song/score_flow
Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)
diffusion-models flax jax neural-odes neurips-2021 normalizing-flows score-based-generative-modeling score-based-generative-models score-matching
Last synced: 20 Nov 2024
https://github.com/flaxengine/flaxsamples
Collection of example projects for Flax Engine
flax flax-engine gamedev samples templates visual-studio
Last synced: 15 Dec 2024
https://github.com/n2cholas/jax-resnet
Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
computer-vision deep-learning flax jax machine-learning neural-networks resnet
Last synced: 18 Nov 2024
https://github.com/young-geng/m3ae_public
Multimodal Masked Autoencoders (M3AE): A JAX/Flax Implementation
computer-vision flax jax natural-language-processing transformers
Last synced: 06 Nov 2024
https://github.com/shyamsn97/hyper-nn
Easy Hypernetworks in Pytorch and Jax
flax hypernetworks jax machine-learning neural-networks pytorch
Last synced: 19 Dec 2024
https://github.com/flaxengine/flaxdocs
Flax Engine Documentation
docfx docs documentation flax flax-documentation flax-engine flaxengine game-engine gamedev
Last synced: 19 Dec 2024
https://github.com/evgenii-nikishin/rl_with_resets
JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"
atari atari2600 deep-learning deep-reinforcement-learning deepmind-control-suite dmc drq flax gym jax machine-learning overfitting reinforcement-learning sac soft-actor-critic spr
Last synced: 14 Dec 2024
https://github.com/borealisai/flora-opt
This is the official repository for the paper "Flora: Low-Rank Adapters Are Secretly Gradient Compressors" in ICML 2024.
deep-learning flax jax large-language-models lora memory-efficient-tuning optax peft random-projection transformers
Last synced: 15 Dec 2024
https://github.com/google-research/jestimator
Amos optimizer with JEstimator lib.
deep-learning flax jax language-model machine-learning mnist optimization optimization-algorithms optimizer transformer
Last synced: 20 Dec 2024
https://github.com/XanaduAI/GradDFT
GradDFT is a JAX-based library enabling the differentiable design and experimentation of exchange-correlation functionals using machine learning techniques.
deep-learning density-functional-theory differentiable-computing flax jax pyscf python quantum-chemistry scientific-computing
Last synced: 13 Nov 2024
https://github.com/xanaduai/graddft
GradDFT is a JAX-based library enabling the differentiable design and experimentation of exchange-correlation functionals using machine learning techniques.
deep-learning density-functional-theory differentiable-computing flax jax pyscf python quantum-chemistry scientific-computing
Last synced: 06 Nov 2024
https://github.com/forhaoliu/instructrl
Instruction Following Agents with Multimodal Transforemrs
flax instruction-following instructions jax machine-learning reinforcement-learning transformer vision-language-transformer
Last synced: 10 Oct 2024
https://github.com/evgenii-nikishin/omd
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
automatic-differentiation deep-learning deep-reinforcement-learning dm-haiku dqn flax gym haiku jax machine-learning model-based-reinforcement-learning model-based-rl reinforcement-learning sac soft-actor-critic
Last synced: 14 Dec 2024
https://github.com/flaxengine/arizonaframework
Ready to use, open-source framework for creating games in Flax
flax flax-engine flaxengine gamedev indiedev scripting
Last synced: 20 Dec 2024
https://github.com/alvarobartt/safejax
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
flax flax-serialization haiku haiku-serialization huggingface jax jax-serialization objax objax-serialization safejax safetensors serialization
Last synced: 14 Nov 2024
https://github.com/ivaniscoding/gnn-for-combinatorial-optimization
JAX + Flax implementation of "Combinatorial Optimization with Physics-Inspired Graph Neural Networks" by Schuetz et al.
combinatorial-optimization deep-learning flax github gnn graph-neural-networks jax learn nbdev optimization qubo
Last synced: 27 Oct 2024
https://github.com/camenduru/stable-diffusion-diffusers-colab
🤗 HuggingFace Diffusers Flax TPU and PyTorch GPU for Colab
colab diffusers discord flax gdrive huggingface huggingface-diffusers pytorch tpu
Last synced: 09 Nov 2024
https://github.com/revsic/jax-variational-diffwave
Jax/Flax implementation of Variational-DiffWave.
diffwave flax jax variational-diffusion-models vdm
Last synced: 30 Nov 2024
https://github.com/nipunsadvilkar/roberta-base-mr
RoBERTa Marathi Language model trained from scratch during huggingface 🤗 x flax community week
flax huggingface jax marathi roberta transformers
Last synced: 08 Nov 2024
https://github.com/matthias-wright/jax-fid
FID computation in Jax/Flax.
fid flax frechet-inception-distance jax
Last synced: 27 Oct 2024
https://github.com/nestordemeure/AdaHessianJax
Jax implementation of the AdaHessian optimizer
adahessian deep-learning flax hessian jax optimizer
Last synced: 09 Nov 2024
https://github.com/flaxengine/arizonaframeworksample
Sample project showcasing usage of Arizona Framework for Flax Engine
flax flaxengine gamedev gameplay
Last synced: 15 Nov 2024
https://github.com/bobmcdear/flaim
Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.
computer-vision deep-learning flax jax machine-learning
Last synced: 27 Nov 2024
https://github.com/flaxengine/fidelityfx-fsr
AMD FidelityFX Super Resolution plugin for Flax Engine
amd fidelityfx flax flax-engine fsr gamedev graphics
Last synced: 15 Nov 2024
https://github.com/flaxengine/imgui
Dear ImGui plugin for Flax Engine that adds debug GUI interface to game viewport.
cpp csharp flax flax-engine imgui
Last synced: 15 Nov 2024
https://github.com/flaxengine/onlineplatformsteam
Online platform implementation of Steam for games made with Flax Engine
flax flax-engine gamedev steam
Last synced: 15 Nov 2024
https://github.com/frankroeder/goal_conditioned_rl
Goal-conditioned reinforcement learning like 🔥
actor-critic deep-deterministic-policy-gradient deep-reinforcement-learning distrax droq flax goal-conditioned-rl goals gymnasium gymnasium-robotics jax mpi4jax optax reinforcement-learning robotics soft-actor-critic
Last synced: 06 Dec 2024
https://github.com/nobuotsukamoto/jax_examples
Jax, Flax, examples (ImageClassification, SemanticSegmentation, and more...)
deep-learning flax jax semantic-segmentation
Last synced: 09 Nov 2024
https://github.com/flaxengine/networksample
Multiplayer sample with players lobby and chat in Flax Engine
csharp flax flax-engine multiplayer networking
Last synced: 15 Nov 2024
https://github.com/ilex-paraguariensis/yerbamate
A framework-agnostic deep learning package and experiment manager
deep-learning flax jax keras machine-learning neural-network package-manager pytorch pytorch-lightning replicability research tensorflow
Last synced: 20 Oct 2024
https://github.com/jessefarebro/flax-mup
Maximal Update Parametrization (μP) with Flax & Optax.
Last synced: 30 Nov 2024
https://github.com/flaxengine/dlss
NVIDIA DLSS plugin for Flax Engine
dlss flax flax-engine gamedev graphics nvidia
Last synced: 15 Nov 2024
https://github.com/ppwwyyxx/dash-docset-jax
Create nice dash/zeal docset for JAX+Flax
Last synced: 13 Oct 2024
https://github.com/sea-snell/jax_v_pytorch
comparing Jax (haiku and flax) to PyTorch on simple DL tasks.
cnn deep-learning flax haiku jax mlp neural-network pytorch
Last synced: 06 Dec 2024
https://github.com/ryu1845/hyena-jax
Implementation of Hyena Hierarchy in JAX
flax hyena jax llm machine-learning nlp python
Last synced: 29 Nov 2024
https://github.com/stefnotch/keyboard-letters-game
A little FlaxEngine remake of https://9gag.com/gag/aer3qNQ
Last synced: 16 Oct 2024
https://github.com/phuijse/tutorial_jax
Tutorial de JAX para entrenar redes neuronales artificiales
flax jax neural-networks tutorial
Last synced: 15 Nov 2024
https://github.com/terrence-ou/generative-deep-learning-2nd-edition-pytorch-jax
Pytorch/JAX implementation of examples in Generative Deep Learning 2nd Edition by David Foster
flax generative-deep-learning jax pytorch
Last synced: 08 Nov 2024
https://github.com/evanatyourservice/llm-jax
Train a mistral-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.
fineweb flax huggingface jax large-language-models llm optimization
Last synced: 28 Nov 2024
https://github.com/shawonashraf/jax_examples
Deep Learning examples using the Jax ecosystem of libraries
computer-vision deep-learning flax jax nlp notebook-jupyter tutorial
Last synced: 15 Dec 2024
https://github.com/flaxengine/exampleplugin
Example plugin project for Flax Engine
Last synced: 15 Nov 2024
https://github.com/erfanzar/fjdiffusion
implementation of in Jax/Flax !
flax generative-adversarial-network generative-model jax stable-diffusion unet
Last synced: 07 Nov 2024
https://github.com/jeongsol-kim/jax_practice
Jax/Flax implementation practice for various neural networks
Last synced: 21 Nov 2024
https://github.com/stefnotch/flax-minesweeper
A smol Minesweeper game using the amazing FlaxEngine
Last synced: 12 Nov 2024
https://github.com/stefnotch/toolboxscriptdragndropplugin
A little demo of how to add custom things to the FlaxEngine toolbox
Last synced: 12 Nov 2024
https://github.com/bcebere/jax_tabular_examples
Various ML models implemented using JAX
flax jax machine-learning neural-network python
Last synced: 14 Nov 2024
https://github.com/flaxengine/onlineplatformxboxlive
Online platform implementation of Xbox Live for games made with Flax Engine
flax flax-engine gamedev xbox xbox-live
Last synced: 15 Nov 2024
https://github.com/flaxengine/bistro
Amazon Lumberyard Bistro for Flax Engine
bistro flax flax-engine game gamedev project
Last synced: 15 Nov 2024
https://github.com/muhd-umer/pvt-flax
Unofficial JAX/Flax implementation of Pyramid Vision Transformer: A Versatile Backbone for Dense Prediction without Convolutions.
deep-learning flax image-classification implementation jax
Last synced: 07 Nov 2024
https://github.com/beegass/vaes
Reproducible code showing the various types of variational autoencoders I have implemented
flax flux jax pytorch variational-autoencoder variational-inference
Last synced: 03 Oct 2024
https://github.com/bobmcdear/open-clip-jax
Implementation of OpenAI's CLIP in JAX/Flax
computer-vision contrastive-learning deep-learning flax jax
Last synced: 27 Nov 2024
https://github.com/evanatyourservice/image-classification-jax
Image classification in JAX with ViT, resnet, cifar10, cifar100, imagenette, and imagenet
flax image-classification jax resnet vision-transformer
Last synced: 28 Nov 2024
https://github.com/stefnotch/flax-bspline-experiment
A little B-Spline implementation for Flax
Last synced: 12 Nov 2024
https://github.com/dirmeier/denoising-diffusion-operators
Denoising diffusion operators in Flax
denoising-diffusion-operator flax jax python score-based-generative-modeling
Last synced: 16 Nov 2024
https://github.com/daniel-j-h/nedem
Neural implicit digital elevation model
dem digital-elevation-model flax glo-30 jax machine-learning optax
Last synced: 18 Dec 2024
https://github.com/dirmeier/consistency-flow-matching
Flow and consistency matching in Flax
consistency-matching flax flow-matching jax python
Last synced: 14 Dec 2024
https://github.com/stefnotch/flax-birthday-cake
A little birthday cake for someone
Last synced: 12 Nov 2024
https://github.com/dirmeier/diffusion-transformer
A diffusion transformer implementation in Flax
diffusion-transformer dit flax jax python
Last synced: 16 Nov 2024
https://github.com/shawonashraf/postagger-lstm-jax
LSTM POS Tagger implementation in Jax and Flax
Last synced: 15 Dec 2024
https://github.com/flaxengine/flaxdocshost
Flax Engine Documentation compiled for static hosting
docfx documentation flax flaxengine
Last synced: 15 Nov 2024
https://github.com/daskol/yax
Yet Another X: JAX/FLAX module tracing, modification, and evaluation.
Last synced: 06 Nov 2024
https://github.com/leliuga/datrin
dataset, train, inference
dataset flax inference jax safetensors sentencepiece train
Last synced: 07 Nov 2024
https://github.com/achrafmam2/rsna-mammo-kaggle
cnn computer-vision deep-learning flax health-ai-hackathon jax kaggle
Last synced: 14 Nov 2024
https://github.com/alexjackson1/tx
A Flax-based library for examining transformers, based on TransformerLens.
deep-learning flax jax transformerlens transformers
Last synced: 20 Dec 2024
https://github.com/alexjackson1/jax-transformer
A re-implementation of Callum McDougall's "Transformer from Scratch" using Flax.
deep-learning flax jax transformer
Last synced: 20 Dec 2024