Awesome Open Source
Search
Programming Languages
Languages
All Categories
Categories
About
Search results for flax
flax
x
81 search results found
Transformers
⭐
124,049
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Pyprobml
⭐
6,066
Python code for "Probabilistic Machine learning" book by Kevin Murphy
Flaxengine
⭐
5,246
Flax Engine – multi-platform 3D game engine
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.
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
Fortuna
⭐
824
A Library for Uncertainty Quantification.
Score_sde
⭐
782
Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)
Long Range Arena
⭐
635
Long Range Arena for Benchmarking Efficient Transformers
Prompt Tuning
⭐
580
Original Implementation of Prompt Tuning from Lester, et al, 2021
Jaxrl
⭐
576
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
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.
Tevatron
⭐
334
Tevatron - A flexible toolkit for neural retrieval research and development.
Commonlooputils
⭐
261
CLU lets you write beautiful training loops in JAX.
Gradcache
⭐
254
Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint
Flaxmodels
⭐
201
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
Orbax
⭐
187
Orbax provides common utility libraries for JAX users.
Jaxseq
⭐
172
Train very large language models in Jax.
Flaxapi
⭐
129
Old repository with C# Editor and C# API for creating games in Flax Engine
Flax
⭐
123
general purpose programming language, in the vein of C++
Efficientnet Jax
⭐
121
EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
Koclip
⭐
117
KoCLIP: Korean port of OpenAI CLIP, in Flax
Flaxdocs
⭐
83
Flax Engine Documentation
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🔥
Jax Models
⭐
82
Unofficial JAX implementations of deep learning research papers
Jestimator
⭐
77
Amos optimizer with JEstimator lib.
Medclip
⭐
75
A multi-modal CLIP model trained on the medical dataset ROCO
Jax Resnet
⭐
67
Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
Hyper Nn
⭐
65
Easy Hypernetworks in Pytorch and Jax
Scifive
⭐
62
SciFive: a text-text transformer model for biomedical literature
M3ae_public
⭐
58
Multimodal Masked Autoencoders (M3AE): A JAX/Flax Implementation
Imgx Diffseg
⭐
53
A JAX-based deep learning framework for image segmentation using diffusion models.
Jax Rl
⭐
52
JAX implementations of core Deep RL algorithms
Graddft
⭐
45
Grad-DFT is a JAX-based library enabling the differentiable design and experimentation of exchange-correlation functionals using machine learning techniques.
Sac N Jax
⭐
36
Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch
Redco
⭐
35
MLSys Workshop NeurIPS 2023 - Redco: A Lightweight Tool to Automate Distributed Training and Inference
Score_flow
⭐
35
Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)
Chef Transformer
⭐
29
Chef Transformer 🍲 .
Stable Diffusion Diffusers Colab
⭐
29
🤗 HuggingFace Diffusers Flax TPU and PyTorch GPU for Colab
Jax Variational Diffwave
⭐
28
Jax/Flax implementation of Variational-DiffWave.
Arizonaframework
⭐
26
Ready to use, open-source framework for creating games in Flax
Language Model
⭐
24
한국어 언어 모델 학습을 위한 프로젝트(Flax, Pytorch with Huggingface Accelerate)
Omd
⭐
23
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
Flaxlm
⭐
23
Train large Huggingface models with flexible data/model parallelism in Flax/JAX
Xpag
⭐
20
a modular reinforcement learning library with JAX agents
Ntg
⭐
19
Generative neural networks for 3D terrain.
Mps Rnn
⭐
18
Code for 'From Tensor Network Quantum States to Tensorial Recurrent Neural Networks'.
Jax Fid
⭐
18
FID computation in Jax/Flax.
Omega
⭐
16
A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environments.
Fidelityfx Fsr
⭐
15
AMD FidelityFX Super Resolution plugin for Flax Engine
Safejax
⭐
13
Serialize JAX, Flax, Haiku, or Objax model params with `safetensors`
Arizonaframeworksample
⭐
13
Sample project showcasing usage of Arizona Framework for Flax Engine
Flaxoptimizers
⭐
13
A collection of optimizers, some arcane others well known, for Flax.
Llmdatadistill
⭐
12
distill large scale web page text
Flaim
⭐
12
Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.
Gans Jax
⭐
12
Implementation of several Generative Adversarial Networks in JAX / Flax
Rl_with_resets
⭐
11
JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"
Ramsey
⭐
10
Probabilistic deep learning using JAX
Yerbamate
⭐
10
A framework-agnostic deep learning package and experiment manager
Jax_trainer
⭐
10
Lightning-like training API for JAX with Flax
Jax So Vits Svc 5.0
⭐
10
A so-vits-svc-5.0 rewritten in jax and flax. Works properly on TPU v2/v3.
Fjformer
⭐
10
Embark on a journey of paralleled/unparalleled computational prowess with FJFormer - an arsenal of custom Jax Flax Functions and Utils that elevate your AI endeavors to new heights!
Onlineplatformsteam
⭐
10
Online platform implementation of Steam for games made with Flax Engine
Helx
⭐
9
Interoperating between (Deep) Reiforcement Learning libraries
Flax_nerf
⭐
9
Unofficial implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis, using Flax with the Linen API
Adahessianjax
⭐
9
Jax implementation of the AdaHessian optimizer
Holographic Reduced Representations
⭐
9
Holographic Reduced Representations
Networksample
⭐
9
Multiplayer sample with players lobby and chat in Flax Engine
Roberta Base Mr
⭐
9
RoBERTa Marathi Language model trained from scratch during huggingface 🤗 x flax community week
Rl Basics
⭐
8
Simple single file implementation of Deep Reinforcement Learning algorithms
Functional Diffusion Processes
⭐
8
Official code for Continuous-Time Functional Diffusion Processes (NeurIPS 2023).
Polyglot Jax Inference
⭐
8
TPU에서 한국어용 LLM 추론을 위한 Jax/Flax 구현체입니다.
Gnn For Combinatorial Optimization
⭐
8
JAX + Flax implementation of "Combinatorial Optimization with Physics-Inspired Graph Neural Networks" by Schuetz et al.
Imgui
⭐
8
Dear ImGui plugin for Flax Engine that adds debug GUI interface to game viewport.
Starcoder Jax
⭐
6
a Jax/Flax inference code of StarCoder
Dlss
⭐
6
NVIDIA DLSS plugin for Flax Engine
Journal
⭐
6
Console for Flax Engine with command handling
Dash Docset Jax
⭐
6
Create nice dash/zeal docset for JAX+Flax
Jax Journey
⭐
6
A pathway and collection of resources to learning Jax from beginning to advance.
Gsoc 22 Tensorflow Resources And Notebooks
⭐
5
GSoC'22 @ TensorFlow Notebooks, Code and much more
Jax_examples
⭐
5
Jax, Flax, examples (ImageClassification, SemanticSegmentation, and more...)
Seed
⭐
5
A cross-platform launcher for the Flax game engine.
1-81 of 81 search results
Privacy
|
About
|
Terms
|
Follow Us On Twitter
Copyright 2018-2024 Awesome Open Source. All rights reserved.