Awesome Open Source
Search
Programming Languages
Languages
All Categories
Categories
About
Search results for jax automatic differentiation
automatic-differentiation
x
jax
x
17 search results found
Pennylane
⭐
2,022
PennyLane is a cross-platform Python library for differentiable programming of quantum computers. Train a quantum computer the same way as a neural network.
Ott
⭐
417
Optimal Transport tools implemented with the JAX framework, to get auto-diff, parallel and jit-able computations.
Tensorcircuit
⭐
230
Tensor network based quantum software framework for the NISQ era
Jaxfluids
⭐
203
Differentiable Fluid Dynamics Package
Adam
⭐
82
adam implements a collection of algorithms for calculating rigid-body dynamics in Jax, CasADi, PyTorch, and Numpy.
Catalyst
⭐
75
A JIT compiler for hybrid quantum programs in PennyLane
Jax_fdm
⭐
74
Auto-differentiable and hardware-accelerated force density method
Jax Fenics Adjoint
⭐
71
Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
Bayex
⭐
50
Bayesian Optimization in JAX
Kerax
⭐
39
Keras-like APIs for JAX framework
Jaxsim
⭐
32
A scalable physics engine and multibody dynamics library implemented with JAX. With JIT batteries 🔋
Jax Fenics
⭐
27
Differentiable interface to FEniCS for JAX
Omd
⭐
23
JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"
Gwfast
⭐
20
A Fisher information matrix python package for GW detector networks.
Autofd
⭐
16
Automatic Functional Differentiation in JAX
Fcontin
⭐
5
Numerical Continuation using just the residual. 5k+ downloads
Autodiff101
⭐
5
An introduction to Automatic Differentiation with theory and code examples.
Related Searches
Python Jax (261)
1-17 of 17 search results
Privacy
|
About
|
Terms
|
Follow Us On Twitter
Copyright 2018-2024 Awesome Open Source. All rights reserved.