jax — 开源软件健康 — inspect.software

PyPI · Go · crates.io

100卓越健康指数

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

Python · Go★ 11.2K↓ 26.6M/月2026年8月27日

MIT2026年8月27日 · 指标 2.10.0

PyPI

98卓越健康指数

PennyLaneAI/pennylane

PennyLane is an open-source quantum software platform for quantum computing, quantum machine learning, and quantum chemistry. Create meaningful quantum algorithms, from inspiration to implementation.

Python★ 3,4042026年8月13日

Apache-2.02026年8月13日 · 指标 2.10.0

PyPI

98卓越健康指数

huggingface/diffusers

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.

Python★ 34.3K↓ 8.3M/月2026年8月12日

Apache-2.02026年8月12日 · 指标 2.10.0

PyPI

98卓越健康指数

keras-team/keras

Deep Learning for humans

Python★ 64.2K2026年8月4日

Apache-2.02026年8月4日 · 指标 2.10.0

PyPI

96卓越健康指数

google/orbax

Orbax provides common checkpointing and persistence utilities for JAX users

Python★ 534↓ 5.9M/月2026年8月29日

Apache-2.02026年8月29日 · 指标 2.10.0

PyPI

96卓越健康指数

iree-org/iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.

C++ · C · MLIR★ 3,9072026年8月28日

Apache-2.02026年8月28日 · 指标 2.10.0

PyPI

96卓越健康指数

keras-team/keras-hub

Pretrained model hub for Keras 3.

Python★ 9892026年9月4日

Apache-2.02026年9月4日 · 指标 2.10.0

PyPI

95卓越健康指数

HIPS/autograd

Efficiently computes derivatives of NumPy code.

Python★ 7,523↓ 5.1M/月2026年8月28日

MIT2026年8月28日 · 指标 2.10.0

PyPI · crates.io

95卓越健康指数

google/flax

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

Jupyter Notebook · Python★ 7,304↓ 4.4M/月2026年8月27日

Apache-2.02026年8月27日 · 指标 2.10.0

PyPI

95卓越健康指数

jax-ml/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Python★ 36.1K2026年8月5日

Apache-2.02026年8月5日 · 指标 2.10.0

PyPI

94卓越健康指数

google/grain

Library for reading and processing ML training data.

Python★ 7522026年7月21日

Apache-2.02026年7月21日 · 指标 2.10.0

PyPI

93卓越健康指数

pyro-ppl/numpyro

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

Python★ 2,7352026年8月13日

Apache-2.02026年8月13日 · 指标 2.10.0

PyPI

90优秀健康指数

patrick-kidger/equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Python★ 2,935↓ 1.1M/月2026年7月30日

Apache-2.02026年7月30日 · 指标 2.10.0

PyPI

88优秀健康指数

google/etils

Collection of eclectic utils for python.

Python · Jupyter Notebook★ 259↓ 8.1M/月2026年8月27日

Apache-2.02026年8月27日 · 指标 2.10.0

PyPI

86优秀健康指数

arogozhnikov/einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

Python · Jupyter Notebook★ 9,582↓ 27.1M/月2026年8月28日

MIT2026年8月28日 · 指标 2.10.0

PyPI

86优秀健康指数

astro-informatics/s2fft

S2FFT: Differentiable and accelerated spherical transforms

Python★ 229↓ 1,179/月2026年8月16日

MIT2026年8月16日 · 指标 2.10.0

PyPI · npm

86优秀健康指数

magenta/magenta-realtime

Magenta RealTime 2: An Open-Weights Live Music Model

Python · C++★ 1,683↓ 817/月2026年7月18日

Apache-2.02026年7月18日 · 指标 2.10.0

PyPI

86优秀健康指数

tensorflow/datasets

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

Python★ 4,5812026年8月28日

Apache-2.02026年8月28日 · 指标 2.10.0

PyPI

83优秀健康指数

google/torchax

torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JAX-Pytorch interoperability, meaning, one can mix JAX & Pytorch syntax together when authoring ML programs, and run it in every hardware JAX can run.

Python★ 231↓ 116.4K/月2026年7月19日

Apache-2.02026年7月19日 · 指标 2.10.0

PyPI

83优秀健康指数