-
optax Public
Forked from google-deepmind/optaxOptax is a gradient processing and optimization library for JAX.
Python Apache License 2.0 UpdatedDec 4, 2024 -
qutip Public
Forked from qutip/qutipQuTiP: Quantum Toolbox in Python
Python BSD 3-Clause "New" or "Revised" License UpdatedSep 18, 2024 -
awesome-jax Public
Forked from n2cholas/awesome-jaxJAX - A curated list of resources https://github.com/google/jax
-
-
diffrax Public
Forked from patrick-kidger/diffraxNumerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
Python Apache License 2.0 UpdatedJan 17, 2024 -
jax Public
Forked from jax-ml/jaxComposable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Python Apache License 2.0 UpdatedJan 7, 2024 -
-
-
pytorch Public
Forked from pytorch/pytorchTensors and Dynamic neural networks in Python with strong GPU acceleration
C++ Other UpdatedMay 16, 2023
{{ message }}