jax

Deep Learning for humans

可组合的 Python+NumPy 程序转换：微分、向量化、JIT 到 GPU/TPU 等功能

互动式深度学习书籍，包含多框架代码、数学与讨论。已在 70 个国家 500 所大学采用，包括 Stanford、MIT、Harvard 与 Cambridge。

🏆 每周更新的机器学习 Python 库排行榜。

从第一原理出发，利用现有研究文献对 Claude Mythos 架构进行理论重建。

在不同框架之间转换机器学习代码

AI 开发者平台，使用 Weights & Biases 进行模型训练与微调，并从实验到部署全流程管理模型。

灵活且强大的张量运算，让代码易读且可靠（支持 pytorch、jax、TensorFlow 等）

高效计算 NumPy 代码的导数

通过这本非传统教材，成为破解 AI/ML 研究员/工程师，涵盖数学、计算与机器学习直观概念。

Flax is a neural network library for JAX that is designed for flexibility.

Kevin Murphy《Probabilistic Machine Learning》一书的 Python 代码

我相信本书主要针对的研究生与求职者将从阅读中受益；同时，我也希望即使是最有经验的研究人员也会觉得它同样引人入胜。

JAX 版 OpenAI Whisper 模型实现，可在 TPU 上提升至 70 倍速度

TFDS 是一套可直接使用于 TensorFlow、Jax 等的资料集集合

科学机器学习与物理知识驱动学习的函数库。

可重定向的 MLIR 基础机器学习编译器与运行时工具包

Scenic：一个用于计算机视觉研究及更广泛应用的Jax库

透过世界模型掌握多领域专业知识

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit and 4-bit floating point (FP8 and FP4) precision on Hopper, Ada and Blackwell GPUs, to provide better performance with lower memory utilization in both training and inference.

PennyLane 是一个用于量子计算、量子机器学习和量子化学的开源量子软件平台。从灵感到实现，创建有意义的量子算法。

基于 JAX 的神经网络库

在加速器硬件上进行大规模并行刚体物理模拟。

阿姆斯特丹大学（MSc AI）2023 秋季深度学习课程的 Jupyter Notebook 教学资料库。

用于信号处理与声音合成的函数式编程语言

一个 Python 工具箱，用于在 PyTorch、TensorFlow 与 JAX 中创建能欺骗神经网络的对抗样本。

优雅易用的神经网络与 JAX 科学计算。 https://docs.kidger.site/equinox/

🔮 一种令人耳目一新的函数式深度学习方法，与您喜爱的库兼容

使用由 JAX 提供自动微分和 JIT 编译至 GPU/TPU/CPU 支持的 NumPy 进行概率编程。

在 JAX 中实现的 蒙特卡洛树搜索。