Unofficial JAX implementations of deep learning research papers
Old repository with C# Editor and C# API for creating games in Flax Engine
KoCLIP: Korean port of OpenAI CLIP, in Flax
EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen...
Official code for "Maximum Likelihood Training of Score-Based Diffusion ...
Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet...
Collection of example projects for Flax Engine
Flax Engine Documentation
Amos optimizer with JEstimator lib.
JAX implementations of core Deep RL algorithms
Instruction Following Agents with Multimodal Transforemrs
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learn...
Single-file SAC-N implementation on jax with flax and equinox. 10x faste...
Train GEMMA on TPU/GPU! (Codebase for training Gemma-Ko Series)
Ready to use, open-source framework for creating games in Flax