FlaxEngine / FlaxEngine

Flax Engine – multi-platform 3D game engine

google / flax

Flax is a neural network library for JAX that is designed for flexibility.

Toni-SM / skrl

Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for Gymnasium/Gym, NVIDIA Isaac Lab, MuJoCo Playground and other environments

google / orbax

Orbax provides common checkpointing and persistence utilities for JAX users

monatis / ggmlc

A multi-framework neural network compiler lowering PyTorch, JAX, Flax, and Keras models to portable, high-performance GGML execution