Awesome Open Source
Search
Programming Languages
Languages
All Categories
Categories
About
Search results for jax flax
flax
x
jax
x
65 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
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.
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
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
Score_flow
⭐
35
Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)
Redco
⭐
35
MLSys Workshop NeurIPS 2023 - Redco: A Lightweight Tool to Automate Distributed Training and Inference
Chef Transformer
⭐
29
Chef Transformer 🍲 .
Jax Variational Diffwave
⭐
28
Jax/Flax implementation of Variational-DiffWave.
Flaxlm
⭐
23
Train large Huggingface models with flexible data/model parallelism in Flax/JAX
Omd
⭐
23
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
Xpag
⭐
20
a modular reinforcement learning library with JAX agents
Ntg
⭐
19
Generative neural networks for 3D terrain.
Jax Fid
⭐
18
FID computation in Jax/Flax.
Mps Rnn
⭐
18
Code for 'From Tensor Network Quantum States to Tensorial Recurrent Neural Networks'.
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.
Safejax
⭐
13
Serialize JAX, Flax, Haiku, or Objax model params with `safetensors`
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
Llmdatadistill
⭐
12
distill large scale web page text
Rl_with_resets
⭐
11
JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"
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.
Ramsey
⭐
10
Probabilistic deep learning using JAX
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!
Yerbamate
⭐
10
A framework-agnostic deep learning package and experiment manager
Adahessianjax
⭐
9
Jax implementation of the AdaHessian optimizer
Roberta Base Mr
⭐
9
RoBERTa Marathi Language model trained from scratch during huggingface 🤗 x flax community week
Holographic Reduced Representations
⭐
9
Holographic Reduced Representations
Flax_nerf
⭐
9
Unofficial implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis, using Flax with the Linen API
Helx
⭐
9
Interoperating between (Deep) Reiforcement Learning libraries
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.
Rl Basics
⭐
8
Simple single file implementation of Deep Reinforcement Learning algorithms
Starcoder Jax
⭐
6
a Jax/Flax inference code of StarCoder
Jax Journey
⭐
6
A pathway and collection of resources to learning Jax from beginning to advance.
Dash Docset Jax
⭐
6
Create nice dash/zeal docset for JAX+Flax
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...)
Related Searches
Python Jax (415)
Deep Learning Jax (98)
Machine Learning Jax (88)
Pytorch Jax (78)
Jupyter Notebook Jax (65)
1-65 of 65 search results
Privacy
|
About
|
Terms
|
Follow Us On Twitter
Copyright 2018-2024 Awesome Open Source. All rights reserved.