Pinned Repositories
active-heave-compensation
awesome-graph-explainability-papers
Papers about explainability of GNNs
bpinns-simple-problems
B-PINNS for dynamical systems.
coax
Modular framework for Reinforcement Learning in python
differentiable-mpc
diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
flax
Flax is a neural network library for JAX that is designed for flexibility.
gru-attitude-control-geodetic-missions
Precise attitude control for Geodetic missions using JAX.
jax-bayes
Bayesian inference with Python and Jax.
shrenikvz's Repositories
shrenikvz/active-heave-compensation
shrenikvz/awesome-graph-explainability-papers
Papers about explainability of GNNs
shrenikvz/bpinns-simple-problems
B-PINNS for dynamical systems.
shrenikvz/coax
Modular framework for Reinforcement Learning in python
shrenikvz/differentiable-mpc
shrenikvz/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
shrenikvz/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
shrenikvz/flax
Flax is a neural network library for JAX that is designed for flexibility.
shrenikvz/gru-attitude-control-geodetic-missions
Precise attitude control for Geodetic missions using JAX.
shrenikvz/jax-bayes
Bayesian inference with Python and Jax.
shrenikvz/jaxpi
shrenikvz/lineax
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
shrenikvz/mpc.pytorch
A fast and differentiable model predictive control (MPC) solver for PyTorch.
shrenikvz/neuromancer
Pytorch-based framework for solving parametric constrained optimization problems, physics-informed system identification, and parametric model predictive control.
shrenikvz/numpyro
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
shrenikvz/pyro
Deep universal probabilistic programming with Python and PyTorch
shrenikvz/shrenikvz
shrenikvz/trajax