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)