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.
  ☆58Updated 6 months ago
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆53Updated 2 years ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆49Updated last week
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆30Updated 5 years ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆45Updated 4 years ago
• crowsonkb / mdmm-jax
  Gradient-based constrained optimization for JAX
  ☆37Updated 3 years ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated 2 years ago
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆55Updated 4 years ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆42Updated 5 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• mfinzi / constrained-hamiltonian-neural-networks
  ☆112Updated 4 years ago
• MJHutchinson / ExtrinsicGaugeIndependentVectorGPs
  A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes
  ☆26Updated 4 years ago
• elizabethnewman / hessQuik
  Computing gradients and Hessians of feed-forward networks with GPU acceleration
  ☆20Updated last year
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 3 years ago
• patrick-kidger / sympytorch
  Turning SymPy expressions into PyTorch modules.
  ☆156Updated 2 years ago
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆28Updated 4 years ago
• mfinzi / residual-pathway-priors
  ☆22Updated 4 years ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 5 years ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated last year
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated 2 years ago
• deep-spin / sparse_continuous_distributions
  This repository provides open-source code for sparse continuous distributions and corresponding Fenchel-Young losses.
  ☆16Updated 2 years ago
• AndreiB137 / EinFields
  Einstein Fields official implementation in JAX.
  ☆69Updated 5 months ago
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆107Updated 2 years ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆72Updated last month
• IBM / pylspack
  A lightweight, multithreaded Python package for sketching, column selection, leverage scores and related computations.
  ☆20Updated 4 months ago
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆131Updated this week
• ASEM000 / kernex
  Stencil computations in JAX
  ☆71Updated 2 months ago
• gehring / fax
  ☆80Updated 4 years ago
• moskomule / hypergrad
  Simple and extensible hypergradient for PyTorch
  ☆18Updated 2 years ago
• google-deepmind / einshape
  ☆107Updated last year