jax

分散式 GPU 加速演化計算框架。包含完整的演化演算法與基準測試問題庫。

讓大型語言模型（LLM）變得簡單，EasyLM 是在 JAX/Flax 中進行 LLM 預訓練、微調、評估與服務的一站式解決方案。

一個簡單、高效能且可擴充的 Jax LLM！

一個可擴充的深度學習庫

在 Kubernetes 上進行分散式 AI 模型訓練與 LLM 微調

JAX - 資源精選清單 https://github.com/google/jax

JAX 中的數值微分方程式求解器。支援自動微分與 GPU運算。https://docs.kidger.site/diffrax/

用於多體位能表示與分子動態模擬的深度學習套件

針對 JAX/NumPy/PyTorch 等陣列形狀與資料型態的型別註解與執行期檢查。https://docs.kidger.site/jaxtyping/

Magenta RealTime 2：一個開源權重的現場音樂模型

基於 Kubernetes 的自動化機器學習

TensorLy：Python 中的張量學習。

此儲存庫提供 AIMv1 與 AIMv2 研究專案的程式碼與模型檢查點。

《How To Scale Your Model》的主頁，這本短篇部落格風格教材介紹了如何在 TPU 上擴充 LLM

使用 Python 的數位音訊工作站；包含 VST 樂器/效果器、參數自動化、FAUST、JAX、Warp 標記與 JUCE 處理器

PDEBench：用於科學機器學習的廣泛基準測試集

💁 自然語言處理 Transformers 模型精選寶庫，包含論文、影片、部落格、官方儲存庫以及 Colab 筆記本。 🛫☑️

基於 JAX 的熱力學超圖模型庫

極速的端對端 Jax RL 實作

模組化強化學習（RL）函式庫（使用 PyTorch、JAX 與 NVIDIA Warp 實作），支援 Gymnasium/Gym、NVIDIA Isaac Lab、MuJoCo Playground 等環境。

JAX 中的硬體加速、可批次化且可微分的最佳化工具。

一個使用 JAX 進行機率狀態空間建模的 Python 套件

Keras 3 的預訓練模型中樞

基於 JAX 的計算流體力學

🦁 適用於 JAX 中多智慧體強化學習快速實驗的研究友善程式碼庫

JavaScript 中的 JAX – 用於網路的 ML 函式庫，執行於 WebGPU 與 Wasm 上

高效能的 ML 模型服務框架，提供動態批次處理與 CPU/GPU 管線，充分發效能。

🕹️ 一套在 JAX 中具備可擴展性的多元強化學習環境套件

用於讀取與處理 ML 訓練資料的程式庫。

使用 JAX 框架實作的最佳化傳輸工具，用於解決各種規模的的大型配對問題。