全部标签
目录标签

#jax

公开记录中带有此标签的全部仓库——标签来自其 GitHub 主题或软件包注册表发布的关键词。健康度量遵循与记录其余部分相同的版本化方法论。

46 条记录
标签为“jax”按健康指数排序
PyPI
75良好健康指数
avitai/artifex
A research-focused modular generative modeling library built on JAX/Flax NNX
Python★ 1↓ 2,893/月2026年8月29日
MIT2026年8月29日 · 指标 2.10.0
PyPI
75良好健康指数
tensorcircuit/tensorcircuit-ng
Next-gen AI-native tensor-network-based quantum software framework
Python★ 87↓ 2,149/月2026年8月22日
Apache-2.02026年8月22日 · 指标 2.10.0
PyPI
73良好健康指数
google-deepmind/distrax
该仓库未发布描述。
Python★ 645↓ 175.1K/月2026年7月30日
Apache-2.02026年7月30日 · 指标 2.10.0
PyPI
71良好健康指数
araffin/sbx
SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms
Python★ 608↓ 1,960/月2026年9月6日
MIT2026年9月6日 · 指标 2.10.0
PyPI · npm
71良好健康指数
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
Python★ 2,890↓ 25.7M/月2026年8月27日
MIT2026年8月27日 · 指标 2.10.0
PyPI
71良好健康指数
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日 · 指标 2.10.0
PyPI
69良好健康指数
google/jaxopt
Hardware accelerated, batchable and differentiable optimizers in JAX.
Python★ 1,0552026年8月4日
Apache-2.02026年8月4日 · 指标 2.10.0
PyPI
63中等健康指数
OpenHCSDev/PolyStore
该仓库未发布描述。
Python★ 0↓ 9,937/月2026年8月22日
MIT2026年8月22日 · 指标 2.10.0
PyPI
63中等健康指数
elkins-lab/diff-biophys
Differentiable biophysical modeling in JAX (SAXS, NMR, CD), with Jupyter Notebook examples.
Python★ 02026年7月17日
MIT2026年7月17日 · 指标 2.10.0
PyPI
59中等健康指数
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日 · 指标 2.10.0
PyPI
59中等健康指数
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日 · 指标 2.10.0
PyPI
57中等健康指数
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日 · 指标 2.10.0
PyPI
57中等健康指数
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日 · 指标 2.10.0
PyPI
57中等健康指数
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日 · 指标 2.10.0
PyPI
51中等健康指数
42Ar/jaxon
Save pytrees efficiently in hdf5 files.
Python★ 112026年8月11日
GPL-3.02026年8月11日 · 指标 2.10.0
PyPI
19危急健康指数
google-deepmind/jraph
A Graph Neural Network Library in Jax
Python★ 1,4732026年8月13日
Apache-2.02026年8月13日 · 指标 2.10.0