flax

Kevin Murphy著『Probabilistic Machine learning』のPythonコード

Flax Engine – マルチプラットフォーム対応の3Dゲームエンジン

アムステルダム大学（MSc AI）のディープラーニングコース（2023年秋）の授業用Jupyterノートブックチュートリアルのリポジトリ

大規模言語モデル（LLM）を簡単に。EasyLM は、JAX/Flax で LLM の事前学習、ファインチューニング、評価、サービングを行うためのワンストップソリューションです。

Gymnasium/Gym、NVIDIA Isaac Lab、MuJoCo Playgroundなどの環境をサポートする、モジュール式強化学習（RL）ライブラリ（PyTorch、JAX、NVIDIA Warpで実装）。

Tevatron - スケール、言語、モダリティを横断する統合ドキュメント検索ツールキット。SIGIR 2023、SIGIR 2025 デモ採択。

Orbax は、JAX ユーザー向けに共通のチェックポイントおよび永続化ユーティリティを提供します