一套 SIMD 加速的数值计算内核库，覆盖距离、点积、矩阵运算和地理几何计算，支持 16 种数值类型与多平台架构，为向量检索和科学计算提供底层高性能算子。
把 Kaldi 语音识别工具包封装成 Python 接口，让你能在 Python 里直接调用 Kaldi 的特征提取、声学模型和语言模型，省去写 C++ 胶水代码的麻烦。
把 NumPy 风格的 Python 代码自动微分、向量化并 JIT 编译到 GPU/TPU，写起来像 NumPy，跑起来像高性能计算。
把 ONNX、PyTorch、Keras、CoreML 等格式的神经网络模型直接拖进浏览器就能看结构图，排查模型转换和层连接问题很方便。
Hugging Face 官方数据集库，汇集海量即用型 AI 数据集，并提供高效的数据加载与处理工具，省去自己找数据、写加载代码的麻烦。
把 PyTorch、TensorFlow、JAX、NumPy 等框架的机器学习代码互相转换，省去为不同框架重写模型的麻烦。
用一套可读的 Einstein 记号统一 PyTorch、JAX、TensorFlow 等框架的张量操作，让 reshape、转置、rearrange 这类代码更清晰、更不容易出错。
提供大量可直接用于 TensorFlow、JAX 等框架的数据集集合，省去自己下载、解析和预处理数据的麻烦，统一接口加载即可训练模型。
跨平台多媒体处理框架，支持 GPU 加速和异构调度，可把转码、AI 推理、算法模块和直播流处理串成一条流水线。
把普通文本渲染成逼真的手写体图片，支持手写字体、信纸背景和字距行距笔画随机扰动，提供 GUI 和 CLI 两种用法。
…u can still build from source for older CPUs. jax.numpy.atleast 1d , jax.numpy.atleast 2d , and jax.numpy.atleast 3d now return tuples of arrays rather than lists, matchi
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
…t/export.html compatibility-guarantees. Added jax.numpy.top k , which implements numpy.top k , added in in NumPy v2.6.0 ( 39729). Breaking changes The exec time optimizat
…ease. No intentional metric changes (the eig→eigh NumPy 2.5 fix shifts metric outputs at float-rounding level, ~2e-6 to 1e-4 relative — documented in the changelog), unli
…atsnew/3.13.html) was implemented. - Support for [Numpy 2](https://numpy.org/devdocs/release/2.0.0-notes.html) was implemented.