Alle Tags
Katalog-Tag

#jax

Alle Repositories im öffentlichen Register, die dieses Tag tragen — aus ihren GitHub-Topics oder den von ihren Paket-Registries veröffentlichten Schlagwörtern. Die Gesundheit wird nach derselben versionierten Methodik gemessen wie im übrigen Register.

24 Einträge
Getaggt als „jax“Geordnet nach Gesundheitsindex
PyPI · Go · crates.io
94ExzellentGesundheitsindex
wandb/wandb
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/Monat20. Juli 2026
MIT20. Juli 2026 · Metriken 1.13.0
PyPI
86ExzellentGesundheitsindex
keras-team/keras
Deep Learning for humans
Python★ 64.2K↓ 18.4M/Monat16. Juli 2026
Apache-2.016. Juli 2026 · Metriken 1.13.0
PyPI
83GutGesundheitsindex
jax-ml/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Python★ 36K16. Juli 2026
Apache-2.016. Juli 2026 · Metriken 1.13.0
PyPI
82GutGesundheitsindex
HIPS/autograd
Efficiently computes derivatives of NumPy code.
Python★ 7.51218. Juli 2026
MIT18. Juli 2026 · Metriken 1.13.0
PyPI
81GutGesundheitsindex
google/grain
Library for reading and processing ML training data.
Python★ 75221. Juli 2026
Apache-2.021. Juli 2026 · Metriken 1.13.0
PyPI · crates.io
78GutGesundheitsindex
google/flax
Flax is a neural network library for JAX that is designed for flexibility.
Jupyter Notebook · Python★ 7.271↓ 4.3M/Monat18. Juli 2026
Apache-2.018. Juli 2026 · Metriken 1.13.0
PyPI
75GutGesundheitsindex
tensorflow/datasets
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
Python★ 4.58221. Juli 2026
Apache-2.021. Juli 2026 · Metriken 1.13.0
PyPI
74GutGesundheitsindex
google/etils
Collection of eclectic utils for python.
Python · Jupyter Notebook★ 257↓ 8.3M/Monat18. Juli 2026
Apache-2.018. Juli 2026 · Metriken 1.13.0
PyPI
72GutGesundheitsindex
arogozhnikov/einops
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
Python · Jupyter Notebook★ 9.54817. Juli 2026
MIT17. Juli 2026 · Metriken 1.13.0
PyPI · npm
72GutGesundheitsindex
magenta/magenta-realtime
Magenta RealTime 2: An Open-Weights Live Music Model
Python · C++★ 1.683↓ 817/Monat18. Juli 2026
Apache-2.018. Juli 2026 · Metriken 1.13.0
PyPI
71GutGesundheitsindex
google-deepmind/chex
Keine Repository-Beschreibung veröffentlicht.
Python★ 948↓ 2M/Monat21. Juli 2026
Apache-2.021. Juli 2026 · Metriken 1.13.0
PyPI
71GutGesundheitsindex
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/Monat19. Juli 2026
Apache-2.019. Juli 2026 · Metriken 1.13.0
PyPI
71GutGesundheitsindex
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/Monat18. Juli 2026
Eigene Lizenz18. Juli 2026 · Metriken 1.13.0
PyPI
68MittelGesundheitsindex
galacticdynamics/phasecurvefit
Construct Paths through Phase-Space Points
Python★ 417. Juli 2026
MIT17. Juli 2026 · Metriken 1.13.0
PyPI
65MittelGesundheitsindex
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/Monat18. Juli 2026
Apache-2.018. Juli 2026 · Metriken 1.13.0
PyPI
62MittelGesundheitsindex
tenax-lab/tenax
JAX-based tensor network library with label-based contraction and symmetry-aware block-sparse tensors
Python★ 719. Juli 2026
Apache-2.019. Juli 2026 · Metriken 1.13.0
PyPI
62MittelGesundheitsindex
tensorcircuit/tensorcircuit-ng
Next-gen AI-native tensor-network-based quantum software framework
Python★ 81↓ 0/Monat13. Juli 2026
Apache-2.013. Juli 2026 · Metriken 1.13.0
PyPI · npm
61MittelGesundheitsindex
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
Python★ 2.890↓ 22.6M/Monat18. Juli 2026
MIT18. Juli 2026 · Metriken 1.13.0
PyPI
60MittelGesundheitsindex
elkins-lab/diff-biophys
Differentiable biophysical modeling in JAX (SAXS, NMR, CD), with Jupyter Notebook examples.
Python★ 017. Juli 2026
MIT17. Juli 2026 · Metriken 1.13.0
PyPI
56MittelGesundheitsindex
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/Monat17. Juli 2026
MIT17. Juli 2026 · Metriken 1.13.0
PyPI
56MittelGesundheitsindex
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★ 117. Juli 2026
MIT17. Juli 2026 · Metriken 1.13.0
PyPI
55MittelGesundheitsindex
HighDimensionalEconLab/nlls_gram
Gram/dual-form Levenberg-Marquardt nonlinear least-squares solvers for JAX/Flax NNX models
Python★ 1↓ 2.882/Monat18. Juli 2026
MIT18. Juli 2026 · Metriken 1.13.0
PyPI
55MittelGesundheitsindex
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★ 45118. Juli 2026
Apache-2.018. Juli 2026 · Metriken 1.13.0
PyPI
55MittelGesundheitsindex
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★ 45118. Juli 2026
Apache-2.018. Juli 2026 · Metriken 1.13.0