OSS Radar
首页
AI 分类
推理与服务5
大模型推理服务 · 向量化 / 重排服务 · 网关与路由
训练与微调6
训练框架 · 参数高效微调 · 后训练 / RLHF
数据工程4
数据集托管 · 数据清洗与处理 · 合成数据
检索与 RAG9
向量数据库 · RAG 框架 · 检索增强引擎
Agent 与编排6
Agent 框架 · 编码 Agent · 电脑 / 浏览器 Agent
评估 · 实验 · MLOps8
模型评测 · Agent 评测 · 实验管理
多模态生成5
图像生成 · 视频生成 · 视觉语言模型
视觉与 3D5
检测 / 分割 · 3D 重建 · 3D 生成
模型工具链8
量化 · 剪枝与蒸馏 · 格式转换
科研与论文实现3
论文复现 · 时序预测 · 科学计算 / 生物医药
开发工具
开发与框架5构建与交付4底层与逆向3
系统工具
系统与文件7图像与影音工具5实用小工具4
效率工具
记录与整理3输入与阅读4
媒体与内容
播放与媒体库4内容处理3
网络工具
访问与代理3连接与传输4
游戏相关
开发与运行2游玩相关2
安全与隐私
攻防与审计3加密与隐私1
硬件与机器人
嵌入式与物联网5机器人与具身智能3
数据与金融
数据库与存储4金融与量化1
探索排行榜开发语言标签
/
周报
本周新增、涨得最快、发了什么版本
对比项目
两个项目并排看 star、维护状态与硬件
数据体检
收录量、完整度缺口、沉睡项目
提交仓库
推荐项目,附上你的使用经验
反馈与纠错
分类错了、介绍不准、链接失效
关于本站
数据来源与分类原则
登录
OSS Radar

按算法工程工作流分类的开源工具库。数据来自 GitHub 公开 API,收录 6,739 个仓库、129 个细分分类。 带 ✦ 的条目有人工撰写的专家点评。

数据最近同步于 2026/11/10 20:04:40 · 本项目仅做聚合与索引,软件版权归原作者所有

浏览

  • 全部仓库
  • AI 分类
  • 排行榜
  • 开发语言
  • 标签与关键词
  • 搜索

关于

  • 关于本站
  • 数据体检
  • 提交仓库
  • 反馈与纠错
  • 偏好设置
  • 我的收藏
  • 我的点评
© 2026 OSS Radar数据每小时级同步 · 内容由算法与人工共同维护
⌘K
「JAX」找到 21 个结果 · 4ms命中词:jax只看有点评
J
jax
jax-ml·Python

把 NumPy 风格的 Python 代码自动微分、向量化并 JIT 编译到 GPU/TPU,写起来像 NumPy,跑起来像高性能计算。

自动微分+4
366020—
jax-ml/jax · 18.14
K
keras
keras-team·Python

面向人类友好的深度学习 API,用 Python 编写,可跑在 TensorFlow、JAX、PyTorch 上,帮开发者快速搭建和训练神经网络。

深度学习+4
644210—
keras-team/keras · 11.47
B
best-of-ml-python
lukasmasuch

每周更新的机器学习 Python 库排行榜,按类别和评分整理优质库,帮你快速找到 AutoML、深度学习、数据可视化等方向的趁手工具。

机器学习库+4
238600—
lukasmasuch/best-of-ml-python · 10.77
I
ivy
unifyai·Python

把 PyTorch、TensorFlow、JAX、NumPy 等框架的机器学习代码互相转换,省去为不同框架重写模型的麻烦。

框架转换+3
141670—
unifyai/ivy · 10.77
E
einops
arogozhnikov·Python

用一套可读的 Einstein 记号统一 PyTorch、JAX、TensorFlow 等框架的张量操作,让 reshape、转置、rearrange 这类代码更清晰、更不容易出错。

张量操作+4
96330—
arogozhnikov/einops · 10.61
O
OpenMythos
kyegomez·Python

从公开研究文献出发,用第一性原理复现 Claude Mythos 架构的理论实现,供研究者拆解其注意力机制与模型设计思路。

模型架构复现+3
149240—
kyegomez/OpenMythos · 10.55
W
wandb
wandb·Python

面向 AI 开发者的实验追踪与模型管理平台,记录训练指标、超参数和数据集版本,支持从实验到生产的全流程协作。

实验追踪+4
113140—
wandb/wandb · 10.37
D
datasets
tensorflow·Python

提供大量可直接用于 TensorFlow、JAX 等框架的数据集集合,省去自己下载、解析和预处理数据的麻烦,统一接口加载即可训练模型。

数据集+4
46110—
tensorflow/datasets · 10.32
S
scenic
google-research·Python

基于 JAX 的计算机视觉研究库,提供注意力机制、Transformer 和 Vision Transformer 等模型实现,方便快速搭建和复现视觉研究实验。

计算机视觉+4
38480—
google-research/scenic · 10.23
I
iree
iree-org·C++

基于 MLIR 的可重定向机器学习编译器与运行时工具包,把 PyTorch、TensorFlow、JAX、ONNX 等模型编译到 CPU、CUDA、ROCm、Vulkan 等多种硬件上运行。

机器学习编译器+4
40090—
iree-org/iree · 10.07
M
markdown-preview-enhanced
shd101wyy·HTML

Atom 编辑器里的 Markdown 实时预览插件,支持 KaTeX/MathJax 公式、Mermaid/PlantUML 图表和 Pandoc 导出,写技术文档时预览效果一步到位。

Markdown 预览+4
44580—
shd101wyy/markdown-preview-enhanced · 9.65
T
trainer
kubeflow·Go

在 Kubernetes 上跑分布式模型训练和大模型微调的框架,把多机多卡任务编排成 K8s 原生作业,省去自己搭训练集群的麻烦。

分布式训练+4
22790—
kubeflow/trainer · 9.65
K
katib
kubeflow·Python

Kubernetes 上的自动机器学习平台,支持超参数调优和神经网络架构搜索,可对接 PyTorch、JAX 等训练框架。项目已弃用,官方建议迁移到 OptimizationJob 和 Kubeflow Trainer。

超参数调优+4
17070—
kubeflow/katib · 9.65
K
kaggle-tpu-lab
ARahim3·Python

在免费 Kaggle TPU v5e-8 上跑 GLM-5.3-Flash 320B MoE 和 Qwen3.8-27B 等前沿大模型,自研 JAX 推理引擎支持 262k 长上下文和前缀缓存,可接入 Claude Code 等编码代理。

TPU推理+4
7700—
ARahim3/kaggle-tpu-lab · 9.33
F
fast-plate-ocr
ankandrew·Python

轻量级车牌文字识别 OCR 模型,支持 Keras、JAX 和 ONNX Runtime,可快速集成到车牌识别流水线中。

车牌识别+3
8060—
ankandrew/fast-plate-ocr · 9.31
T
Treasure-of-Transformers
ashishpatel26·Jupyter Notebook

汇总 Transformer 自然语言处理模型的论文、视频、博客、官方代码和 Colab 笔记本,方便快速查找和上手各类预训练模型。

Transformer+4
11860—
ashishpatel26/Treasure-of-Transformers · 9.28
M
mosec
mosecorg·Python

高性能机器学习模型服务框架,支持动态批处理和 CPU/GPU 流水线,把单机算力充分利用起来,适合部署 CV、LLM 等模型。

模型服务+4
9040—
mosecorg/mosec · 9.18
J
JetStream
AI-Hypercomputer·Python

面向 XLA 设备(先支持 TPU,后续扩展 GPU)的大模型推理引擎,主打高吞吐和低显存占用,适合在 JAX 生态里跑 Llama、Gemma 等模型。

推理引擎+4
4730—
AI-Hypercomputer/JetStream · 9.09
T
tessera
zengxiao-he·Python

从零实现的大模型蒸馏与推理服务引擎,包含自定义 Triton/CUDA 算子、FSDP 蒸馏、分页 KV 连续批处理、投机解码和 Rust 网关,适合研究推理加速与可解释性。

推理引擎+4
5560—
zengxiao-he/tessera · 9.05
I
i1
zlab-princeton·Python

一套完全开源的强文本到图像扩散模型训练配方,提供 JAX 和 PyTorch 实现,方便复现和二次开发。

文本到图像+4
2940—
zlab-princeton/i1 · 8.93
S
scaling-book
jax-ml·SCSS

一本博客风格的短教材,讲怎么在 TPU 上扩展大模型训练与推理,用 roofline 分析帮你算清算力、显存和通信的瓶颈。

TPU+4
15250—
jax-ml/scaling-book · 5.48

发布说明里提到「JAX」

来自 6 个版本
  • jax-ml/jaxjax-v0.12.0.12026-10-22

    Bug fixes Built C++-generated intrinsic LLVM IR on Windows with clang-cl instead of emitting empty bitcode, enabling vectorized CPU intrinsics on Windows (https

  • jax-ml/jaxjax-v0.12.02026-10-19

    New features Added jax.scipy.special.erfcinv for the inverse complementary error function. Added jax.lax.polynomial for polynomial evaluation. The polynomial primitive is

  • jax-ml/jaxjax-v0.11.22026-09-17

    New features Added jax.numpy.minmax (and jnp.minmax ), which returns both the minimum and maximum of an array, matching NumPy 2.3+ ( 40089). Added jax.lax.log2 and primit

  • jax-ml/jaxjax-v0.11.12026-08-17

    …es Added an error check for trying to deserialize JAX exports that are older than the backwards compatibility window. Without this check the deserialization of expired ar

  • jax-ml/jaxjax-v0.11.02026-07-16

    …g custom derivative rules with the experimental hijax API ( hijax-custom-derivatives ), along with jax.experimental.hijax helpers for deriving VJPHiPrimitive autodiff rul

  • vwxyzjn/cleanrlv1.0.02022-11-14

    … variants, support for google's new ML framework [JAX](https://github.com/google/jax), hyperparameter tuning utilities, and more. CleanRL has come a long way making high-