Alternatives and similar repositories for veni:

Users that are interested in veni are comparing it to the libraries listed below

• elizabethnewman / hessQuik
  Computing gradients and Hessians of feed-forward networks with GPU acceleration
  ☆18Updated last year
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆27Updated 3 years ago
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆48Updated last year
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆51Updated 2 weeks ago
• f-dangel / einconv
  Convolutions and more as einsum for PyTorch
  ☆16Updated 10 months ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆45Updated last week
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆125Updated 7 months ago
• afqueiruga / StatefulOdeNets
  Refining continuous-in-depth neural networks
  ☆39Updated 3 years ago
• liuyao12 / ConvNets-PDE-perspective
  an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets
  ☆22Updated last year
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆44Updated 4 years ago
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆49Updated 3 years ago
• IvanYashchuk / jax-fenics
  Differentiable interface to FEniCS for JAX
  ☆53Updated 3 years ago
• pnkraemer / probfindiff
  Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…
  ☆11Updated last year
• mfinzi / constrained-hamiltonian-neural-networks
  ☆107Updated 3 years ago
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆37Updated 2 years ago
• luca-heltai / numerical-analysis-2021-2022
  ☆14Updated 2 years ago
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆83Updated last month
• zalandoresearch / JaxRK
  RKHS feature vectors, operators, and statistical models using JAX for automatic differentiation
  ☆8Updated 3 years ago
• DiffEqML / diffeqml-research
  This repository contains code released by DiffEqML Research
  ☆90Updated 3 years ago
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆94Updated last year
• tk-rusch / coRNN
  Official code for Coupled Oscillatory RNN (ICLR 2021, Oral)
  ☆43Updated 3 years ago
• MaxGhi8 / HyperNOs
  ☆14Updated this week
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 3 months ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆70Updated last year
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆30Updated 4 years ago
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆14Updated 5 months ago
• jamesvuc / jax-bayes
  Bayesian inference with Python and Jax.
  ☆32Updated 2 years ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆46Updated 3 weeks ago
• samholt / NeuralLaplace
  Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.
  ☆76Updated last week