FlaxEngine/OnlinePlatformSteam

Online platform implementation of Steam for games made with Flax Engine

Language: C++ - Size: 699 KB - Last synced at: 5 months ago - Pushed at: 5 months ago - Stars: 17 - Forks: 4

MasterSkepticista/gpt2

Training GPT-2 on FineWeb-Edu in JAX/Flax

Language: Python - Size: 104 KB - Last synced at: 3 months ago - Pushed at: 8 months ago - Stars: 2 - Forks: 0

flyingblackshark/jax-fcpe

fcpe jax flax version

Language: Python - Size: 29.3 KB - Last synced at: 20 days ago - Pushed at: 6 months ago - Stars: 0 - Forks: 0

Beomi/Gemma-EasyLM

Train GEMMA on TPU/GPU! (Codebase for training Gemma-Ko Series)

Language: Python - Size: 410 KB - Last synced at: 4 months ago - Pushed at: over 1 year ago - Stars: 47 - Forks: 10

seungju-k1m/jax-offpolicy-rl

Jax-Based Off-Policy RL Algorithms

Language: Python - Size: 370 KB - Last synced at: 6 months ago - Pushed at: 6 months ago - Stars: 2 - Forks: 0

perrin-isir/xpag

a modular reinforcement learning library with JAX agents

Language: Python - Size: 5.24 MB - Last synced at: 6 months ago - Pushed at: 6 months ago - Stars: 22 - Forks: 5

FlaxEngine/DLSS

NVIDIA DLSS plugin for Flax Engine

Language: C++ - Size: 34.2 KB - Last synced at: 5 months ago - Pushed at: 11 months ago - Stars: 12 - Forks: 4

google-research/long-range-arena 📦

Long Range Arena for Benchmarking Efficient Transformers

Language: Python - Size: 130 KB - Last synced at: 6 months ago - Pushed at: over 1 year ago - Stars: 747 - Forks: 85

MahmudulAlam/Holographic-Reduced-Representations

Holographic Reduced Representations

Language: Python - Size: 343 KB - Last synced at: 4 months ago - Pushed at: 10 months ago - Stars: 25 - Forks: 5

flyingblackshark/jax-nsf-hifigan

Language: Python - Size: 4.88 KB - Last synced at: 6 months ago - Pushed at: 6 months ago - Stars: 0 - Forks: 0

affjljoo3581/polyglot-jax-inference

TPU에서 한국어용 LLM 추론을 위한 Jax/Flax 구현체입니다.

Language: Python - Size: 9.77 KB - Last synced at: 6 months ago - Pushed at: about 2 years ago - Stars: 12 - Forks: 2

dirmeier/denoising-diffusion-operators

Denoising diffusion operators in Flax

Language: Python - Size: 2.58 MB - Last synced at: 6 months ago - Pushed at: about 1 year ago - Stars: 2 - Forks: 0

BeeGass/VAEs

Reproducible code showing the various types of variational autoencoders I have implemented

Language: Python - Size: 62.6 MB - Last synced at: about 20 hours ago - Pushed at: over 3 years ago - Stars: 3 - Forks: 0

evanatyourservice/llm-jax

Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.

Language: Python - Size: 3.57 MB - Last synced at: 5 months ago - Pushed at: 6 months ago - Stars: 17 - Forks: 1

alvarobartt/safejax

Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`

Language: Python - Size: 748 KB - Last synced at: 3 months ago - Pushed at: over 1 year ago - Stars: 44 - Forks: 5

affjljoo3581/deit3-jax

Jax/Flax implementation of DeiT and DeiT-III (ViT)

Language: Shell - Size: 53.7 KB - Last synced at: 6 months ago - Pushed at: 9 months ago - Stars: 16 - Forks: 6

FlaxEngine/FidelityFX-FSR

AMD FidelityFX Super Resolution plugin for Flax Engine

Language: C - Size: 13 MB - Last synced at: 3 months ago - Pushed at: 5 months ago - Stars: 23 - Forks: 4

vballoli/vit-flax

Implementation of Vision Transformers in Flax

Language: Python - Size: 5.74 MB - Last synced at: 22 days ago - Pushed at: almost 5 years ago - Stars: 18 - Forks: 2

JesseFarebro/flax-mup

Maximal Update Parametrization (μP) with Flax & Optax.

Language: Python - Size: 400 KB - Last synced at: 4 months ago - Pushed at: over 1 year ago - Stars: 10 - Forks: 0

nestordemeure/AdaHessianJax

Jax implementation of the AdaHessian optimizer

Language: Python - Size: 104 KB - Last synced at: 5 months ago - Pushed at: over 4 years ago - Stars: 20 - Forks: 3

yonesuke/RationalNets

JAX/Flax implementation of rational neural nets

Language: Python - Size: 34.2 KB - Last synced at: 22 days ago - Pushed at: almost 3 years ago - Stars: 2 - Forks: 0

evanatyourservice/xLSTM-JAX

An implementation of mLSTM from xLSTM in JAX

Language: Python - Size: 8.79 KB - Last synced at: 3 months ago - Pushed at: 9 months ago - Stars: 3 - Forks: 0

flyingblackshark/jax-bs-roformer

bs-roformer from Music-Source-Seperation

Language: Python - Size: 111 KB - Last synced at: 9 months ago - Pushed at: 9 months ago - Stars: 7 - Forks: 0

keith-murray/ctrnn-jax

Implementation of CT-RNNs in Python's JAX ecosystem

Language: Jupyter Notebook - Size: 1.72 MB - Last synced at: 7 months ago - Pushed at: 7 months ago - Stars: 1 - Forks: 0

novaia/ntg

Generative neural networks for 3D terrain.

Language: Python - Size: 51.4 MB - Last synced at: 9 months ago - Pushed at: 9 months ago - Stars: 33 - Forks: 1

Vadimbuildercxx/mamba_fast_simple

Minimal and efficient JAX implementation of the Mamba State Space Model in JAX/Flax. Inspired by 'Mamba: Linear-Time Sequence Modeling with Selective State Spaces,' this repo provides fast, scalable, and well-documented state-of-the-art sequence modeling tools.

Language: Python - Size: 10.7 KB - Last synced at: 9 months ago - Pushed at: 9 months ago - Stars: 0 - Forks: 0

daskol/yax

Yet Another X: JAX/FLAX module tracing, modification, and evaluation.

Language: Python - Size: 273 KB - Last synced at: 1 day ago - Pushed at: 9 months ago - Stars: 0 - Forks: 0

phlippe/jax_trainer

Lightning-like training API for JAX with Flax

Language: Python - Size: 218 KB - Last synced at: 9 months ago - Pushed at: 9 months ago - Stars: 35 - Forks: 2

Raffaelbdl/kitae

Yet another RL library built with FLAX modules

Language: Python - Size: 4.84 MB - Last synced at: 3 months ago - Pushed at: about 1 year ago - Stars: 1 - Forks: 0

kasinadhsarma/RL-Flex

The RL-Flex project is dedicated to the development and implementation of advanced reinforcement learning algorithms and methodologies.

Language: Python - Size: 289 KB - Last synced at: 9 days ago - Pushed at: 10 months ago - Stars: 0 - Forks: 1

leliuga/datrin

dataset, train, inference

Language: Python - Size: 8.44 MB - Last synced at: 4 days ago - Pushed at: over 1 year ago - Stars: 4 - Forks: 0

haoliuhl/instructrl

Instruction Following Agents with Multimodal Transforemrs

Language: Python - Size: 191 KB - Last synced at: 5 months ago - Pushed at: almost 3 years ago - Stars: 52 - Forks: 5

flyingblackshark/jax-mel-band-roformer

Language: Python - Size: 42 KB - Last synced at: 10 months ago - Pushed at: 10 months ago - Stars: 1 - Forks: 0

BirkhoffG/explainax 📦

JAX-based Model Explanation and Interpretation Library

Language: Jupyter Notebook - Size: 439 KB - Last synced at: 2 months ago - Pushed at: over 1 year ago - Stars: 1 - Forks: 0

minhquoc0712/nnx_speed_benchmark

Language: Python - Size: 3.91 KB - Last synced at: 10 months ago - Pushed at: 10 months ago - Stars: 1 - Forks: 1

FlaxEngine/FlaxDocsHost

Flax Engine Documentation compiled for static hosting

Language: HTML - Size: 672 MB - Last synced at: 6 months ago - Pushed at: 11 months ago - Stars: 0 - Forks: 3

mzguntalan/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.

Language: Python - Size: 127 MB - Last synced at: 5 months ago - Pushed at: about 3 years ago - Stars: 7 - Forks: 2

flyingblackshark/jax-Diffusion-svc-v2

Language: Python - Size: 61.5 MB - Last synced at: 12 months ago - Pushed at: 12 months ago - Stars: 0 - Forks: 0

dogeplusplus/flax-u2net

JAX/Flax implementation of U-2 Net for salient object detection.

Language: Python - Size: 727 KB - Last synced at: 9 months ago - Pushed at: over 3 years ago - Stars: 4 - Forks: 1

flyingblackshark/jax-so-vits-svc

so-vits-svc rewritten in jax and flax.

Language: Python - Size: 16.8 MB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 0 - Forks: 0

stefnotch/flax-minesweeper

A smol Minesweeper game using the amazing FlaxEngine

Language: C# - Size: 299 KB - Last synced at: 6 months ago - Pushed at: almost 4 years ago - Stars: 2 - Forks: 1

stefnotch/keyboard-letters-game

A little FlaxEngine remake of https://9gag.com/gag/aer3qNQ

Language: C# - Size: 394 KB - Last synced at: 5 months ago - Pushed at: over 6 years ago - Stars: 5 - Forks: 1

prabhudavidsheryl/flax_nnx_gpt2

A FLAX NNX implementation of GPT2

Language: Jupyter Notebook - Size: 35.2 KB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 0 - Forks: 0

phuijse/tutorial_jax

Tutorial de JAX para entrenar redes neuronales artificiales

Language: Jupyter Notebook - Size: 15.9 MB - Last synced at: 20 days ago - Pushed at: over 2 years ago - Stars: 6 - Forks: 0

yschimpf/bioflax

bioflax provides a JAX implementation of biologically plausible learning algorithms

Language: Python - Size: 30.6 MB - Last synced at: 1 day ago - Pushed at: over 1 year ago - Stars: 8 - Forks: 2

evgenii-nikishin/omd

JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"

Language: Jupyter Notebook - Size: 268 KB - Last synced at: 24 days ago - Pushed at: about 4 years ago - Stars: 43 - Forks: 5

forgi86/jax-ident

An experimental code base for system identification with Jax.

Language: Python - Size: 138 KB - Last synced at: 7 days ago - Pushed at: about 1 year ago - Stars: 4 - Forks: 0

bcebere/jax_tabular_examples

Various ML models implemented using JAX

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

lweitkamp/GANs-JAX

Implementation of several Generative Adversarial Networks in JAX / Flax

Language: Jupyter Notebook - Size: 1.56 MB - Last synced at: about 1 year ago - Pushed at: over 3 years ago - Stars: 30 - Forks: 5

SafeRoboticsLab/KLGame

Repository for "Blending Data-Driven Priors in Dynamic Games" - RSS 2024

Language: Python - Size: 9.01 MB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 2 - Forks: 0

nissymori/JAX-CORL

Clean single-file implementation of offline RL algorithms in JAX

Language: Python - Size: 423 KB - Last synced at: about 1 year ago - Pushed at: over 1 year ago - Stars: 35 - Forks: 0

azzeddineCH/flash-nanoGPT

Jax/Flax re-write of @karpathy 🐐 NanoGPT using some of the common Jax libraries/features (shmap, pallas, jmp, optax, orbax)

Language: Python - Size: 138 KB - Last synced at: about 1 year ago - Pushed at: about 1 year ago - Stars: 3 - Forks: 1

cqsl/mps-rnn

Code for 'From Tensor Network Quantum States to Tensorial Recurrent Neural Networks'.

Language: Python - Size: 31.3 KB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 20 - Forks: 2

FookieMonster/transformer-lm-japanese

This is a JAX/Flax-based transformer language model trained on a Japanese dataset. It is based on the official Flax example code (lm1b).

Language: Python - Size: 849 KB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 2 - Forks: 0

ozencgungor/CMB_Topology_ML_JAX

Neural network layers and networks to study the topology of the universe. Based on graph convolutions and written in the jax environment

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

giulio98/functional-diffusion-processes

Official code for Continuous-Time Functional Diffusion Processes (NeurIPS 2023).

Language: Python - Size: 4.58 MB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 14 - Forks: 1

FlaxCommunityProjects/FlaxCollaborate

Language: C# - Size: 181 KB - Last synced at: 5 days ago - Pushed at: over 4 years ago - Stars: 12 - Forks: 0

Lemon-cmd/diffusion-jax

Diffusion Probabilistic Model in Jax

Language: Python - Size: 96.7 MB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 5 - Forks: 0

HeegyuKim/language-model

한국어 언어 모델 학습을 위한 프로젝트(Flax, Pytorch with Huggingface Accelerate)

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

revsic/jax-variational-diffwave

Jax/Flax implementation of Variational-DiffWave.

Language: Python - Size: 161 KB - Last synced at: 4 months ago - Pushed at: over 3 years ago - Stars: 40 - Forks: 6

FlaxEngine/ExamplePlugin

Example plugin project for Flax Engine

Language: Smalltalk - Size: 9.77 KB - Last synced at: 6 months ago - Pushed at: 11 months ago - Stars: 4 - Forks: 6

astanziola/siren-flax

SIREN neural networks in Flax

Language: Jupyter Notebook - Size: 1.4 MB - Last synced at: over 1 year ago - Pushed at: over 4 years ago - Stars: 7 - Forks: 1

satojkovic/vit-jax-flax

Vision Transformer from scratch (JAX/Flax).

Language: Jupyter Notebook - Size: 88.9 KB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 6 - Forks: 2

manishemirani/NLP-lib

Unfinished library for the NLP tasks using Jax and flax

Language: Python - Size: 59.6 KB - Last synced at: over 1 year ago - Pushed at: almost 3 years ago - Stars: 0 - Forks: 0

affjljoo3581/starcoder-jax

a Jax/Flax inference code of StarCoder

Language: Python - Size: 18.6 KB - Last synced at: 6 months ago - Pushed at: about 2 years ago - Stars: 12 - Forks: 1

FlaxCommunityProjects/flax-auto-import-resources-plugin

Automatically imports assets

Language: C# - Size: 1.01 MB - Last synced at: over 1 year ago - Pushed at: over 3 years ago - Stars: 3 - Forks: 0

rsanchezmo/jaxer

Jax and Flax Time Series Prediction Transformer

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

ksquarekumar/whisper-stream

Whisper Transcription Service

Language: Jupyter Notebook - Size: 6.21 MB - Last synced at: over 1 year ago - Pushed at: almost 2 years ago - Stars: 0 - Forks: 0

justinphan3110/SciFive

SciFive: a text-text transformer model for biomedical literature

Language: Jupyter Notebook - Size: 15.5 MB - Last synced at: over 1 year ago - Pushed at: over 2 years ago - Stars: 84 - Forks: 14

arasgungore/transformers Fork of huggingface/transformers

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

Language: Python - Size: 177 MB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 2 - Forks: 0

vcharraut/rl-basics

Simple single file implementation of Deep Reinforcement Learning algorithms

Language: Python - Size: 3.25 MB - Last synced at: over 1 year ago - Pushed at: almost 2 years ago - Stars: 11 - Forks: 4

ginofft/flax-BLT

FLAX Implementation of Biodirectional Layout Transformer

Language: Python - Size: 11.3 MB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 5 - Forks: 0

ShawonAshraf/postagger-lstm-jax

LSTM POS Tagger implementation in Jax and Flax

Language: Python - Size: 22.5 KB - Last synced at: 5 months ago - Pushed at: over 1 year ago - Stars: 0 - Forks: 0

nothingTVatYT/FlaxPluginScanner

A scanner for Flax plugins in github repositories

Language: C# - Size: 19.5 KB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 1 - Forks: 0

nothingTVatYT/YAPC

Yet another player controller (rigidbody, physics based FPS) for the Flax Engine

Language: C# - Size: 2.91 MB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 0 - Forks: 0

ilex-paraguariensis/yerbamate

A framework-agnostic deep learning package and experiment manager

Language: Jupyter Notebook - Size: 41.9 MB - Last synced at: 5 days ago - Pushed at: about 2 years ago - Stars: 9 - Forks: 1

BigBalloon8/Science_Bacc

SQA scottish Science Baccalaureate of Christopher Rae. Improving the preformance of distributed data parallelism of low wifi bandwidths

Language: TeX - Size: 1.9 MB - Last synced at: over 1 year ago - Pushed at: over 2 years ago - Stars: 0 - Forks: 0

fabianwinter93/JAX

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

ppwwyyxx/dash-docset-jax

Create nice dash/zeal docset for JAX+Flax

Language: Python - Size: 466 KB - Last synced at: 18 days ago - Pushed at: almost 2 years ago - Stars: 7 - Forks: 2

Lando-L/tsmixer-weather-forecast

Kedro project of time-series-forecast of the multivariate weather using TSMixer.

Language: Python - Size: 20.5 KB - Last synced at: over 1 year ago - Pushed at: almost 2 years ago - Stars: 1 - Forks: 0

aillaud/VQVAE_Flax

Implementation of basic autoencodeur, VAE and VQVAE in Flax

Language: Jupyter Notebook - Size: 1.24 MB - Last synced at: almost 2 years ago - Pushed at: almost 2 years ago - Stars: 0 - Forks: 0

gcucurull/maml_flax

Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.

Language: Python - Size: 5.86 KB - Last synced at: 3 months ago - Pushed at: almost 5 years ago - Stars: 19 - Forks: 0

TimCargan/chemise

A lightweight wrapper with standard boiler plate code to train flax models

Language: Python - Size: 298 KB - Last synced at: 2 months ago - Pushed at: 7 months ago - Stars: 1 - Forks: 0

Ryu1845/hyena-jax

Implementation of Hyena Hierarchy in JAX

Language: Python - Size: 27.3 KB - Last synced at: 4 months ago - Pushed at: over 2 years ago - Stars: 10 - Forks: 2

Sea-Snell/jax_v_pytorch

comparing Jax (haiku and flax) to PyTorch on simple DL tasks.

Language: Python - Size: 157 KB - Last synced at: 5 months ago - Pushed at: about 3 years ago - Stars: 7 - Forks: 0

1kaiser/Snow-cover-area-estimation

Estimating Snow ❄️using NDSI From MODIS satellite Data 2000 to 2021

Language: Jupyter Notebook - Size: 252 MB - Last synced at: over 1 year ago - Pushed at: over 1 year ago - Stars: 0 - Forks: 1

FlaxEngine/FlaxAPI 📦

Old repository with C# Editor and C# API for creating games in Flax Engine

Language: C# - Size: 11.9 MB - Last synced at: almost 2 years ago - Pushed at: over 4 years ago - Stars: 134 - Forks: 29

flyingblackshark/jax-Diffusion-svc

DIffusion SVC rewritten by JAX

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

alexjackson1/tx

A Flax-based library for examining transformers, based on TransformerLens.

Language: Python - Size: 1.13 MB - Last synced at: 5 months ago - Pushed at: over 1 year ago - Stars: 0 - Forks: 1

andylolu2/jax-diffusion

Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.

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

FlaxCommunityProjects/flax-custom-visject-plugin

A custom Visject Plugin Demo for the FlaxEngine

Language: C# - Size: 2.38 MB - Last synced at: over 1 year ago - Pushed at: over 4 years ago - Stars: 2 - Forks: 0

FlaxCommunityProjects/FlaxCommunityProjects.github.io

FlaxEngine Community Projects

Size: 1.95 KB - Last synced at: over 1 year ago - Pushed at: about 6 years ago - Stars: 1 - Forks: 0

Memorytaco/Melt

Another site generator Melt.

Language: Scheme - Size: 273 KB - Last synced at: over 1 year ago - Pushed at: about 6 years ago - Stars: 2 - Forks: 0

stefnotch/flax-birthday-cake

A little birthday cake for someone

Language: C# - Size: 26.6 MB - Last synced at: about 2 months ago - Pushed at: over 6 years ago - Stars: 1 - Forks: 0

stefnotch/ToolboxScriptDragNDropPlugin

A little demo of how to add custom things to the FlaxEngine toolbox

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

stefnotch/flax-bspline-experiment

A little B-Spline implementation for Flax

Language: C# - Size: 1020 KB - Last synced at: 15 days ago - Pushed at: almost 7 years ago - Stars: 2 - Forks: 0

muhd-umer/pvt-flax

Unofficial JAX/Flax implementation of Pyramid Vision Transformer: A Versatile Backbone for Dense Prediction without Convolutions.

Language: Python - Size: 149 KB - Last synced at: 4 months ago - Pushed at: over 1 year ago - Stars: 2 - Forks: 0

EastTower16/LLMDataDistill

distill large scale web page text

Language: C++ - Size: 1.5 MB - Last synced at: about 2 years ago - Pushed at: about 2 years ago - Stars: 12 - Forks: 1

flyingblackshark/jax-MelPE

Audio Mel to F0

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

konstantinos-p/wide_resnet_wfixup_jax

An implementation of WideResNets with Fixup initialization in Jax/Flax. This can be useful for use cases where Batch Normalization should be avoided (for example when using the Laplace approximation to the Bayesian posterior).

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