Alternatives and similar repositories for veni:

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

• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆43Updated last week
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆124Updated 4 months ago
• elizabethnewman / hessQuik
  Computing gradients and Hessians of feed-forward networks with GPU acceleration
  ☆18Updated 11 months ago
• yhtang / FunFact
  Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.
  ☆58Updated 2 years ago
• luca-heltai / numerical-analysis-2021-2022
  ☆14Updated last year
• IvanYashchuk / jax-fenics
  Differentiable interface to FEniCS for JAX
  ☆52Updated 3 years ago
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆48Updated last year
• ermongroup / Neural-PDE-Solver
  ☆15Updated 4 years ago
• atzberg / gmls-nets
  PyTorch implementation of GMLS-Nets. Machine learning methods for scattered unstructured data sets. Methods for learning differential op…
  ☆25Updated last year
• NeilGirdhar / tjax
  Tools for JAX
  ☆44Updated this week
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 3 years ago
• moskomule / hypergrad
  Simple and extensible hypergradient for PyTorch
  ☆16Updated last year
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆27Updated 3 years ago
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆71Updated 4 months ago
• ketch / nodepy
  Analysis of initial value ODE solvers
  ☆80Updated 4 months ago
• google-research / gpax
  ☆28Updated 4 months ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆57Updated last year
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆92Updated last year
• mathLab / ATHENA
  ATHENA: Advanced Techniques for High dimensional parameter spaces to Enhance Numerical Analysis
  ☆51Updated last year
• IBM / pylspack
  A lightweight, multithreaded Python package for sketching, column selection, leverage scores and related computations.
  ☆19Updated 4 months ago
• zalandoresearch / JaxRK
  RKHS feature vectors, operators, and statistical models using JAX for automatic differentiation
  ☆8Updated 3 years ago
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆30Updated 4 years ago
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆49Updated 3 years ago
• brandstetter-johannes / LPSDA
  Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"
  ☆49Updated last year
• mfinzi / constrained-hamiltonian-neural-networks
  ☆102Updated 3 years ago
• mazy1998 / Solving-Optimization-Problems-with-JAX
  Solving Optimization Problems with JAX, code and PDF
  ☆15Updated 4 years ago
• kazukiosawa / ngd_in_wide_nn
  simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…
  ☆14Updated 4 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
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆63Updated 6 months ago