Ecosyste.ms: Packages
An open API service providing package, version and dependency metadata of many open source software ecosystems and registries.
conda-forge.org "jax" keyword
diffrax 0.2.2 💰
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable.9 versions - Latest release: over 1 year ago - 794 stars on GitHub
jaxtyping 0.2.8
Type annotations and runtime checking for shape and dtype of JAX arrays, and PyTrees.2 versions - Latest release: over 1 year ago - 1 dependent package - 438 stars on GitHub
thinc 8.1.5
A refreshing functional take on deep learning, compatible with your favorite libraries45 versions - Latest release: over 1 year ago - 2 dependent packages - 63 dependent repositories - 2,684 stars on GitHub
d2l 0.17.6
Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 400 u...2 versions - Latest release: over 1 year ago - 16,875 stars on GitHub
flax 0.6.1
Flax is a neural network library for JAX that is designed for flexibility.11 versions - Latest release: over 1 year ago - 4 dependent packages - 5 dependent repositories - 4,130 stars on GitHub
jaxopt 0.5.5
Hardware accelerated, batchable and differentiable optimizers in JAX2 versions - Latest release: over 1 year ago - 1 dependent package - 1 dependent repositories - 586 stars on GitHub
jax-cfd 0.2.0
Computational Fluid Dynamics in JAX1 version - Latest release: almost 2 years ago - 439 stars on GitHub
netket 3.5.2
NetKet is an open-source project delivering cutting-edge methods for the study of many-body quant...23 versions - Latest release: over 1 year ago - 422 stars on GitHub
brax 0.0.15
Massively parallel rigidbody physics simulation on accelerator hardware.1 version - Latest release: over 1 year ago - 1,570 stars on GitHub
Top 1.6% on conda-forge.org
68 versions - Latest release: over 1 year ago - 24 dependent packages - 101 dependent repositories - 86,720 stars on GitHub
transformers 4.24.0
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.68 versions - Latest release: over 1 year ago - 24 dependent packages - 101 dependent repositories - 86,720 stars on GitHub
Top 6.5% on conda-forge.org
6 versions - Latest release: about 2 years ago - 11 dependent packages - 36 dependent repositories - 6,526 stars on GitHub
einops 0.4.1
Flexible and powerful tensor operations for readable and reliable code. Supports numpy, pytorch, ...6 versions - Latest release: about 2 years ago - 11 dependent packages - 36 dependent repositories - 6,526 stars on GitHub
foolbox 3.3.1 💰
Foolbox is a Python library that let's you easily run adversarial attacks against machine learnin...5 versions - Latest release: about 3 years ago - 1 dependent package - 2,435 stars on GitHub
tensorly 0.7.0
TensorLy is a Python library that aims at making tensor learning simple and accessible. It allows...6 versions - Latest release: over 2 years ago - 3 dependent packages - 4 dependent repositories - 1,377 stars on GitHub
equinox 0.9.2 💰
Callable PyTrees and filtered JIT/grad transformations => neural networks in JAX.16 versions - Latest release: over 1 year ago - 1 dependent package - 1,049 stars on GitHub
deepxde 1.7.1
DeepXDE is a library for scientific machine learning and physics-informed learning.49 versions - Latest release: over 1 year ago - 1 dependent repositories - 1,540 stars on GitHub
Top 3.3% on conda-forge.org
51 versions - Latest release: over 1 year ago - 23 dependent packages - 37 dependent repositories - 22,285 stars on GitHub
jaxlib 0.3.25
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, an...51 versions - Latest release: over 1 year ago - 23 dependent packages - 37 dependent repositories - 22,285 stars on GitHub
Top 2.7% on conda-forge.org
90 versions - Latest release: over 1 year ago - 33 dependent packages - 60 dependent repositories - 22,297 stars on GitHub
jax 0.3.25
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, an...90 versions - Latest release: over 1 year ago - 33 dependent packages - 60 dependent repositories - 22,297 stars on GitHub
eagerpy 0.29.0 💰
PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code3 versions - Latest release: over 3 years ago - 1 dependent package - 671 stars on GitHub
tensorwaves 0.4.9
TensorWaves is a fitter package that optimizes mathematical models to data samples. The models ca...13 versions - Latest release: over 1 year ago - 5 stars on GitHub
pennylane 0.23.0
PennyLane is a cross-platform Python library for differentiable programming of quantum computers....7 versions - Latest release: almost 2 years ago - 4 dependent packages - 2 dependent repositories - 1,703 stars on GitHub
pennylane-qchem 0.17.0
PennyLane is a cross-platform Python library for differentiable programming of quantum computers....1 version - Latest release: about 2 years ago - 1,703 stars on GitHub
funsor 0.4.3
Funsor is a tensor-like library for functions and distributions. See [Functional tensors for prob...1 version - Latest release: over 1 year ago - 216 stars on GitHub
mpi4jax 0.3.10
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python âš¡29 versions - Latest release: over 1 year ago - 1 dependent package - 263 stars on GitHub
tree-math 0.1.0
Mathematical operations for JAX pytrees1 version - Latest release: almost 2 years ago - 1 dependent package - 140 stars on GitHub
pyhf 0.7.0
The HistFactory p.d.f. template [CERN-OPEN-2012-016] is per-se independent of its implementation ...11 versions - Latest release: over 1 year ago - 1 dependent package - 3 dependent repositories - 242 stars on GitHub
jraph 0.0.1.dev
Jraph (pronounced "giraffe") is a lightweight library for working with graph neural networks in j...1 version - Latest release: over 2 years ago - 1 dependent package - 1,121 stars on GitHub
dm-haiku 0.0.8
Haiku is a simple neural network library for JAX developed by some of the authors of Sonnet, a ne...6 versions - Latest release: over 1 year ago - 1 dependent package - 2 dependent repositories - 2,405 stars on GitHub
autoray 0.5.1
Abstract your array operations.9 versions - Latest release: over 1 year ago - 3 dependent packages - 2 dependent repositories - 95 stars on GitHub
tensor-sensor 0.1.3
TensorSensor clarifies exceptions by augmenting messages and visualizing Python code to indicate ...2 versions - Latest release: over 2 years ago - 720 stars on GitHub
numpyro 0.10.1
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.13 versions - Latest release: over 1 year ago - 1 dependent package - 10 dependent repositories - 1,706 stars on GitHub
Related Keywords
deep-learning
15
tensorflow
14
pytorch
14
machine-learning
13
python
12
numpy
8
neural-networks
4
tensor
3
optimization
3
keras
3
mxnet
3
natural-language-processing
3
quantum
3
hacktoberfest
3
neural-network
3
reinforcement-learning
2
physics-simulation
2
pyro
2
probabilistic-programming
2
strawberryfields
2
quantum-machine-learning
2
quantum-computing
2
quantum-chemistry
2
autograd
2
automatic-differentiation
2
qml
2
qiskit
2
differentiable-computing
2
cupy
2
cirq
2
nlp
2
histfactory
1
dask
1
lazy
1
debugging
1
matrix
1
tracing
1
vector
1
bayesian-inference
1
hmc
1
inference-algorithms
1
sympy
1
optimizer
1
minimizer
1
cas
1
tensorflow2
1
eager-execution
1
scientific-machine-learning
1
pinn
1
physics-informed-learning
1
pde
1
paddle
1
mcmc
1
variational-method
1
high-energy-physics
1
hep-ex
1
hep
1
frequentist-statistics
1
cls
1
closember
1
asymptotic-formulas
1
scientific-computations
1
scikit-hep
1
tree-math
1
xla
1
parallel-computing
1
mpi
1
jit
1
high-performance-computing
1
gpu
1
scipy
1
symbolic
1
statistical-inference
1
statistics
1
graph-neural-networks
1
deep-neural-networks
1
array
1
unitaryhack
1
quantum-state-tomography
1
monte-carlo-methods
1
markov-chain-monte-carlo
1
machine-learning-algorithms
1
hamiltonian
1
exact-diagonalization
1
complex-neural-network
1
cfd
1
recommender-system
1
notebook
1
kaggle
1
hyperparameter-optimization
1
gaussian-processes
1
data-science
1
computer-vision
1
book
1
type-checking
1
spacy
1
machine-learning-library
1
functional-programming
1
artificial-intelligence
1
ai
1