GitHub topics: flax

Repositories

oelin/tape

A tiny (8 line) parameter tape implementation.

Language: Python - Size: 26.4 KB - Last synced at: 3 months ago - Pushed at: about 2 years ago - Stars: 3 - Forks: 0

thevasudevgupta/biobigbird

BigBird for bio-medical domain

Language: Python - Size: 203 KB - Last synced at: about 2 years ago - Pushed at: about 2 years ago - Stars: 1 - Forks: 1

hr0nix/omega

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

Language: Python - Size: 577 KB - Last synced at: about 2 years ago - Pushed at: almost 3 years ago - Stars: 32 - Forks: 4

chef-transformer/chef-transformer

Chef Transformer 🍲 .

Language: Python - Size: 15.3 MB - Last synced at: about 2 years ago - Pushed at: about 3 years ago - Stars: 50 - Forks: 22

MahanFathi/HJxB

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

Language: Python - Size: 142 KB - Last synced at: about 2 years ago - Pushed at: over 3 years ago - Stars: 12 - Forks: 0

manifest/flax-extra

The package provides extra flexibility to Flax using ideas originated at Trax.

Language: Python - Size: 149 KB - Last synced at: about 7 hours ago - Pushed at: almost 4 years ago - Stars: 1 - Forks: 0

Howuhh/sac-n-jax

Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch

Language: Python - Size: 531 KB - Last synced at: over 2 years ago - Pushed at: over 2 years ago - Stars: 36 - Forks: 1

massquantity/tdfo

Common practices for distributed training using various backends

Language: Python - Size: 62.5 KB - Last synced at: almost 2 years ago - Pushed at: almost 2 years ago - Stars: 0 - Forks: 0

the-good-fellas/flan-t5-ptbr

Tutorial para treino de um modelo baseado Flan-T5 usando Flax no GCP-TPU

Language: Python - Size: 183 KB - Last synced at: over 2 years ago - Pushed at: over 2 years ago - Stars: 2 - Forks: 0

ridwan-salau/jax_sentiment_classifier

This repository implements sentiment classifier with Google Jax using Bert transformer as backbone. It also shows model checkpointing and loading for inference.

Language: Python - Size: 6.84 KB - Last synced at: over 2 years ago - Pushed at: over 2 years ago - Stars: 0 - Forks: 0

andyehrenberg/flaxlm

Train large Huggingface models with flexible data/model parallelism in Flax/JAX

Language: Python - Size: 299 KB - Last synced at: over 2 years ago - Pushed at: over 2 years ago - Stars: 24 - Forks: 2

lukaemon/minGPT-reimplemented

An even more minimal re-implementation of minGPT

Language: Python - Size: 128 KB - Last synced at: over 2 years ago - Pushed at: almost 3 years ago - Stars: 2 - Forks: 0

henry-prior/jax-rl

JAX implementations of core Deep RL algorithms

Language: Python - Size: 354 KB - Last synced at: over 2 years ago - Pushed at: over 3 years ago - Stars: 62 - Forks: 9

mayureshagashe2105/GSoC-22-TensorFlow-Resources-and-Notebooks

GSoC'22 @ TensorFlow Notebooks, Code and much more

Language: Jupyter Notebook - Size: 137 MB - Last synced at: over 2 years ago - Pushed at: almost 3 years ago - Stars: 8 - Forks: 1

Crawcik/FlaxPluginManager

Unofficial flax plugin manager will help you add plugins so you dont have to search for them :)

Language: C# - Size: 92.8 KB - Last synced at: over 2 years ago - Pushed at: almost 3 years ago - Stars: 5 - Forks: 0

Sauler/vscode-flax-snippets 📦

All snippets for Flax development

Size: 25.4 KB - Last synced at: over 2 years ago - Pushed at: about 7 years ago - Stars: 2 - Forks: 1

Crawcik/AnthillNet 📦

Universal .NET Networking.

Language: C# - Size: 183 KB - Last synced at: over 2 years ago - Pushed at: over 4 years ago - Stars: 3 - Forks: 0

ossan-tamago/jaxU-Net

U-Net written in Jax and Flax.

Language: Python - Size: 6.84 KB - Last synced at: over 2 years ago - Pushed at: over 2 years ago - Stars: 0 - Forks: 0

gerdm/elements-smc

Language: Jupyter Notebook - Size: 7.37 MB - Last synced at: 5 months ago - Pushed at: over 2 years ago - Stars: 1 - Forks: 0

achrafmam2/rsna-mammo-kaggle

Language: Jupyter Notebook - Size: 2.93 MB - Last synced at: 6 months ago - Pushed at: over 2 years ago - Stars: 0 - Forks: 0

n2cholas/progan-flax

Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation

Language: Python - Size: 292 KB - Last synced at: over 2 years ago - Pushed at: over 4 years ago - Stars: 12 - Forks: 0

myagues/flax_nerf

Unofficial implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis, using Flax with the Linen API

Language: Jupyter Notebook - Size: 141 KB - Last synced at: about 2 years ago - Pushed at: almost 4 years ago - Stars: 12 - Forks: 2

dslisleedh/NCNet-flax

Unofficial implementation of NCNet using flax and jax

Language: Python - Size: 131 KB - Last synced at: over 2 years ago - Pushed at: over 2 years ago - Stars: 0 - Forks: 0

nestordemeure/flaxOptimizers

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

Language: Python - Size: 59.6 KB - Last synced at: over 2 years ago - Pushed at: about 4 years ago - Stars: 28 - Forks: 1

srigas/PINNs

Repository with notebooks about Physics Informed Neural Networks, written in JAX + Flax.

Language: Jupyter Notebook - Size: 20.2 MB - Last synced at: over 2 years ago - Pushed at: almost 3 years ago - Stars: 1 - Forks: 1

deterministic-algorithms-lab/Jax-Journey

A pathway and collection of resources to learning Jax from beginning to advance.

Language: Jupyter Notebook - Size: 597 KB - Last synced at: over 2 years ago - Pushed at: over 4 years ago - Stars: 11 - Forks: 0

FlaxCommunityProjects/FlaxUnitTesting

Simple flax engine unit testing

Language: C# - Size: 37.1 KB - Last synced at: about 2 years ago - Pushed at: almost 6 years ago - Stars: 2 - Forks: 0

teonsen/Flax

FlaUI wrapper(UIA3 only), adding Java and OpenCV feature

Language: C# - Size: 23.4 KB - Last synced at: over 2 years ago - Pushed at: over 5 years ago - Stars: 2 - Forks: 0

n1ghtf4l1/melodious-matilda

Identify a wide variety of bird vocalizations in soundscape recordings

Language: Jupyter Notebook - Size: 17.6 KB - Last synced at: over 1 year ago - Pushed at: about 3 years ago - Stars: 0 - Forks: 0

chris-tng/sasrec

Implementation of Self-Attentive Sequential Recommendation (SASRec) in flax

Language: Python - Size: 16.7 MB - Last synced at: over 2 years ago - Pushed at: over 4 years ago - Stars: 4 - Forks: 0

FlaxCommunityProjects/FlaxVR

Basic VR Module for Flax

Language: C# - Size: 453 KB - Last synced at: about 2 years ago - Pushed at: over 5 years ago - Stars: 6 - Forks: 0

daniel-j-h/nedem

Neural implicit digital elevation model

Language: Python - Size: 1.19 MB - Last synced at: over 1 year ago - Pushed at: over 3 years ago - Stars: 1 - Forks: 0

dantruonghtno1/TPU-tutorial

TPU tutorial, jax, prompt, huggingface

Language: Python - Size: 41 KB - Last synced at: over 2 years ago - Pushed at: about 3 years ago - Stars: 0 - Forks: 0

STomoya/learning-jax

For my learning to train deep models with JAX.

Language: Python - Size: 22.5 KB - Last synced at: over 2 years ago - Pushed at: over 3 years ago - Stars: 0 - Forks: 0

manishemirani/DEQ-Mnist

Implementing Deep Equilibrium model on the Mnist dataset using Jax and Flax

Language: Python - Size: 172 KB - Last synced at: over 2 years ago - Pushed at: over 3 years ago - Stars: 0 - Forks: 0

yonetaniryo/jax-vs-pytorch

Comparisons between PyTorch and JAX (Flax)

Language: Jupyter Notebook - Size: 28.3 KB - Last synced at: 6 months ago - Pushed at: almost 4 years ago - Stars: 4 - Forks: 0

TaijiMonster/plots-move

Automated yet simple and easy plots moving/transfer script with discord notification, you don't have to stare at your screen or keep checking for transfer status :)

Language: Shell - Size: 99.6 KB - Last synced at: over 2 years ago - Pushed at: almost 4 years ago - Stars: 2 - Forks: 0

HitLuca/flaxharvestgraph Fork of stolk/chiaharvestgraph

Graphs the activity of a flax harvester in a linux terminal.

Language: C - Size: 399 KB - Last synced at: over 2 years ago - Pushed at: almost 4 years ago - Stars: 0 - Forks: 0

flax 244 jax 191 deep-learning 62 machine-learning 42 flax-engine 34 pytorch 24 python 23 transformers 19 transformer 18 reinforcement-learning 15 computer-vision 15 huggingface 15 optax 13 nlp 12 gamedev 11 llm 10 tensorflow 10 neural-networks 8 vision-transformer 8 flaxengine 8 language-model 7 tpu 7 neural-network 6 haiku 6 deep-reinforcement-learning 6 sac 6 diffusion-models 6 csharp 5 tutorial 5 soft-actor-critic 5 large-language-models 5 nnx 5 gpt 5 gpt2 4 natural-language-processing 4 gym 4 score-based-generative-modeling 4 diffusion 4 attention 4 kaggle 4 optimization 3 flax-linen 3 image-classification 3 optimizer 3 dm-haiku 3 gnn 3 graph-neural-networks 3 td3 3 language-models 3 colab 3 ai 3 numpyro 3 game-engine 3 transfer-learning 3 offline-reinforcement-learning 3 bert 3 numpy 3 resnet 3 pretrained-models 3 image-generation 3 generative-ai 3 gpt-2 3 deepmind 2 continuous-control 2 deep-deterministic-policy-gradient 2 scientific-computing 2 score-matching 2 gymnasium 2 pytorch-lightning 2 d4rl 2 wandb 2 contrastive-learning 2 attention-mechanism 2 nlp-library 2 clm 2 actor-critic 2 information-retrieval 2 digital-elevation-model 2 rl 2 ml 2 networking 2 model-based-rl 2 model-based-reinforcement-learning 2 dqn 2 cnn 2 gpu-acceleration 2 unfinished 2 vae 2 bayesian-inference 2 roberta 2 score-based-generative-models 2 vision 2 documentation 2 cpp 2 segmentation 2 docfx 2 keras 2 neural-radiance-fields 2 image-processing 2 discord 2