PyPI · Go · crates.io
wandb/wandb
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.
MIT2026年8月27日 · 指标 2.10.0
全部标签
目录标签
#jax
公开记录中带有此标签的全部仓库——标签来自其 GitHub 主题或软件包注册表发布的关键词。健康度量遵循与记录其余部分相同的版本化方法论。
68 条记录
PyPI
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.
Apache-2.02026年9月17日 · 指标 2.10.0
PyPI
keras-team/keras
Deep Learning for humans
Apache-2.02026年9月20日 · 指标 2.10.0
PyPI
google/orbax
Orbax provides common checkpointing and persistence utilities for JAX users
Apache-2.02026年10月21日 · 指标 2.10.0
PyPI
jax-ml/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Apache-2.02026年9月16日 · 指标 2.10.0
PyPI
iree-org/iree
A retargetable MLIR-based machine learning compiler and runtime toolkit.
Apache-2.02026年10月20日 · 指标 2.10.0
PyPI
keras-team/keras-hub
Pretrained model hub for Keras 3.
Apache-2.02026年10月21日 · 指标 2.10.0
PyPI
HIPS/autograd
Efficiently computes derivatives of NumPy code.
MIT2026年8月28日 · 指标 2.10.0
PyPI · crates.io
google/flax
Flax is a neural network library for JAX that is designed for flexibility.
Apache-2.02026年8月27日 · 指标 2.10.0
PyPI
google/grain
Library for reading and processing ML training data.
Apache-2.02026年9月12日 · 指标 2.10.0
PyPI
pyro-ppl/numpyro
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Apache-2.02026年9月17日 · 指标 2.10.0
PyPI
tue-bmd/zea
zea: A Toolbox for Cognitive Ultrasound Imaging
Apache-2.02026年10月1日 · 指标 2.10.0
PyPI
nstarman/quax
Multiple dispatch over abstract array types in JAX.
Apache-2.02026年10月4日 · 指标 2.10.0
PyPI
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Apache-2.02026年9月14日 · 指标 2.10.0
PyPI
GalacticDynamics/coordinax
Coordinates in JAX
MIT2026年10月16日 · 指标 2.10.0
PyPI
GalacticDynamics/unxt
Unitful Quantities in JAX
BSD-3-Clause2026年9月18日 · 指标 2.10.0
PyPI
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.
Apache-2.02026年9月11日 · 指标 2.10.0
PyPI
GW-JAX-Team/ripple
Differentiable gravitational waveforms in JAX
MIT2026年10月14日 · 指标 2.10.0
PyPI
instadeepai/jumanji
🕹️ A diverse suite of scalable reinforcement learning environments in JAX
Apache-2.02026年10月3日 · 指标 2.10.0
PyPI
thomaspinder/GPJax
Gaussian processes in JAX and Equinox.
MIT2026年9月19日 · 指标 2.10.0
PyPI
google/etils
Collection of eclectic utils for python.
Apache-2.02026年8月27日 · 指标 2.10.0
PyPI
google-deepmind/dm-haiku
JAX-based neural network library
Apache-2.02026年9月19日 · 指标 2.10.0
PyPI
arogozhnikov/einops
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
MIT2026年8月28日 · 指标 2.10.0
PyPI
tensorflow/datasets
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
Apache-2.02026年8月28日 · 指标 2.10.0
PyPI
GW-JAX-Team/flowMC
Normalizing-flow enhanced sampling package for probabilistic inference in JAX
MIT2026年10月12日 · 指标 2.10.0
PyPI
chaobrain/brainstate
A State-based Transformation System for Brain Modeling.
Apache-2.02026年9月18日 · 指标 2.10.0
PyPI
exoplanet-dev/jaxoplanet
Astronomical time series analysis with JAX
MIT2026年10月19日 · 指标 2.10.0
PyPI · npm
magenta/magenta-realtime
Magenta RealTime 2: An Open-Weights Live Music Model
Apache-2.02026年9月10日 · 指标 2.10.0
PyPI
GalacticDynamics/quax-blocks
Blocks for constructing `quax` classes
BSD-3-Clause2026年10月20日 · 指标 2.10.0
PyPI
GalacticDynamics/xmmutablemap
Immutable Map, compatible with Jax
MIT2026年10月11日 · 指标 2.10.0