jax

使用 JAX 的可微分有限元素法

Tevatron：跨規模、語言與模態的統一文件檢索工具包。於 SIGIR 2023、SIGIR 2025 展示。

用於車牌文字辨識的輕量級快速 OCR 模型

機器學習指南。學習所有關於機器學習工具、函式庫、框架、大型語言模型（LLMs）以及模型訓練的知識。

多體量子系統的機器學習演算法

物理資訊機器學習教學（Pytorch 與 Jax）

JAX 與 Equinox 中的高斯過程。

♟️ 基於 JAX 的向量化強化學習遊戲環境

可微分流體力學套件

JAX+Equinox 中的非線性最佳化（求根、最小平方法等）。https://docs.kidger.site/optimistix/

JAX 與 Equinox 中的線性求解器。https://docs.kidger.site/lineax

使用基於物理資訊的神經網路 (FBPINNs) 求解與偏微分方程相關的正向與反向問題

關於 Google Cloud TPU 你想知道的一切

Felafax 正在為非 NVIDIA GPU 建構 AI 基礎設施

從教師模型到模型切片——從零開始的 LLM 蒸餾與服務引擎：包含自訂 Triton/CUDA 核心、FSDP 蒸餾、paged-KV 連續批次處理、推測解碼、Rust 閘道、JAX 預測器與可解釋性工具。

Pax 是一個基於 Jax 的機器學習框架，用於訓練大規模模型。Pax 支援進階且完全可配置的實驗與平行化，並展現了業界領先的模型 FLOP 利用率。

JAX 陣列的零複製 MPI 通訊，為 Python 中的高效能運算 (HPC) 應用加速。

Orbax 為 JAX 使用者提供通用的檢查點與持久化工具

Yahoo Open-NSFW 模型的 Keras 實作。

https://srush.github.io/annotated-s4 的實作

基於單剛體模型並以 Python 編寫的四足機器人模型預測控制器。支援基於梯度（acados）或基於採樣（jax）的方法。

XLB：用於物理導向機器學習的加速格子波茲曼方法 (Accelerated Lattice Boltzmann)