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

PyPI · Go · crates.io

94优秀健康指数

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

Python · Go★ 11.2K↓ 23M/月2026年7月20日

MIT2026年7月20日 · 指标 1.13.0

PyPI

86优秀健康指数

keras-team/keras

Deep Learning for humans

Python★ 64.2K↓ 18.4M/月2026年7月16日

Apache-2.02026年7月16日 · 指标 1.13.0

PyPI

83良好健康指数

jax-ml/jax

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

Python★ 36K2026年7月16日

Apache-2.02026年7月16日 · 指标 1.13.0

PyPI

82良好健康指数

HIPS/autograd

Efficiently computes derivatives of NumPy code.

Python★ 7,5122026年7月18日

MIT2026年7月18日 · 指标 1.13.0

PyPI

81良好健康指数

google/grain

Library for reading and processing ML training data.

Python★ 7522026年7月21日

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

PyPI · crates.io

78良好健康指数

google/flax

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

Jupyter Notebook · Python★ 7,271↓ 4.3M/月2026年7月18日

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

PyPI

75良好健康指数

tensorflow/datasets

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

Python★ 4,5822026年7月21日

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

PyPI

74良好健康指数

google/etils

Collection of eclectic utils for python.

Python · Jupyter Notebook★ 257↓ 8.3M/月2026年7月18日

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

PyPI

72良好健康指数

arogozhnikov/einops

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

Python · Jupyter Notebook★ 9,5482026年7月17日

MIT2026年7月17日 · 指标 1.13.0

PyPI · npm

72良好健康指数

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日 · 指标 1.13.0

PyPI

71良好健康指数

google-deepmind/chex

该仓库未发布描述。

Python★ 948↓ 2M/月2026年7月21日

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

PyPI

71良好健康指数

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日 · 指标 1.13.0

PyPI

71良好健康指数

patrick-kidger/jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

Python★ 1,842↓ 6.4M/月2026年7月18日

自定义许可证2026年7月18日 · 指标 1.13.0

PyPI

68中等健康指数

galacticdynamics/phasecurvefit

Construct Paths through Phase-Space Points

Python★ 42026年7月17日

MIT2026年7月17日 · 指标 1.13.0

PyPI

65中等健康指数

tillahoffmann/jax-mps

A JAX backend for Apple Metal Performance Shaders (MPS), enabling GPU-accelerated JAX computations on Apple Silicon.

C++ · Python★ 192↓ 3,492/月2026年7月18日

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

PyPI

62中等健康指数

tenax-lab/tenax

JAX-based tensor network library with label-based contraction and symmetry-aware block-sparse tensors

Python★ 72026年7月19日

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

PyPI

62中等健康指数

tensorcircuit/tensorcircuit-ng

Next-gen AI-native tensor-network-based quantum software framework

Python★ 81↓ 0/月2026年7月13日

Apache-2.02026年7月13日 · 指标 1.13.0

PyPI · npm

61中等健康指数

explosion/thinc

🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

Python★ 2,890↓ 22.6M/月2026年7月18日

MIT2026年7月18日 · 指标 1.13.0

PyPI

60中等健康指数

elkins-lab/diff-biophys

Differentiable biophysical modeling in JAX (SAXS, NMR, CD), with Jupyter Notebook examples.

Python★ 02026年7月17日

MIT2026年7月17日 · 指标 1.13.0

PyPI

56中等健康指数

elkins-lab/diff-ensemble

Differentiable VAE framework for predicting protein structural ensembles (IDPs) consistent with SAXS and NMR data. Built on JAX/Flax.

Python★ 1↓ 158/月2026年7月17日

MIT2026年7月17日 · 指标 1.13.0

PyPI

56中等健康指数

elkins-lab/resonance-flow

JAX-native differentiable protein folding framework integrating experimental NMR constraints (RDCs) and biophysical "self-correction." Several Jupyter Notebooks visualize the concepts.

Python★ 12026年7月17日

MIT2026年7月17日 · 指标 1.13.0

PyPI

55中等健康指数

HighDimensionalEconLab/nlls_gram

Gram/dual-form Levenberg-Marquardt nonlinear least-squares solvers for JAX/Flax NNX models

Python★ 1↓ 2,882/月2026年7月18日

MIT2026年7月18日 · 指标 1.13.0

PyPI

55中等健康指数

ai-hypercomputer/jetstream

JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).

Python★ 4512026年7月18日

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

PyPI

55中等健康指数

google/jetstream

JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).

Python★ 4512026年7月18日

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

#jax