Alternatives and similar repositories for veni:

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

• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆44Updated 4 years ago
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆123Updated 6 months ago
• yhtang / FunFact
  Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.
  ☆58Updated 2 years ago
• IvanYashchuk / jax-fenics
  Differentiable interface to FEniCS for JAX
  ☆53Updated 3 years ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆45Updated this week
• 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.
  ☆48Updated last month
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆77Updated 2 weeks ago
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆93Updated last year
• elizabethnewman / hessQuik
  Computing gradients and Hessians of feed-forward networks with GPU acceleration
  ☆18Updated last year
• vtopt / qnstop
  Quasi-Newton Algorithm for Stochastic Optimization
  ☆10Updated 2 years ago
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆27Updated 3 years ago
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆30Updated 4 years ago
• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆67Updated 2 years ago
• mfinzi / constrained-hamiltonian-neural-networks
  ☆106Updated 3 years ago
• juliagusak / neural-ode-metasolver
  Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561
  ☆24Updated 4 years ago
• f-dangel / phd-thesis
  Source code for my PhD thesis: Backpropagation Beyond the Gradient
  ☆20Updated 2 years ago
• mfinzi / residual-pathway-priors
  ☆18Updated 3 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• IBM / pylspack
  A lightweight, multithreaded Python package for sketching, column selection, leverage scores and related computations.
  ☆19Updated 6 months ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 4 years ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• MJHutchinson / ExtrinsicGaugeIndependentVectorGPs
  A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes
  ☆25Updated 3 years ago
• ermongroup / SPN_Variational_Inference
  PyTorch implementation for "Probabilistic Circuits for Variational Inference in Discrete Graphical Models", NeurIPS 2020
  ☆17Updated 3 years ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆44Updated this week
• luca-heltai / numerical-analysis-2021-2022
  ☆14Updated last year
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆39Updated 4 years ago
• IPL-UV / jaxkern
  A lightweight didactic library of kernel methods using the back-end JAX.
  ☆12Updated 2 years ago
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆37Updated 2 years ago