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
  ☆45Updated 4 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆49Updated last month
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 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
• MJHutchinson / ExtrinsicGaugeIndependentVectorGPs
  A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes
  ☆25Updated 3 years ago
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆30Updated 5 years ago
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆49Updated 3 years ago
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆49Updated last year
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆27Updated 3 years ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆55Updated 2 months ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 6 months ago
• patrick-kidger / sympytorch
  Turning SymPy expressions into PyTorch modules.
  ☆148Updated last year
• moskomule / hypergrad
  Simple and extensible hypergradient for PyTorch
  ☆18Updated 2 years ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆40Updated 4 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
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆71Updated last year
• mlysy / rodeo-legacy
  Probabilistic Solution of Differential Equations
  ☆13Updated 3 years ago
• f-dangel / einconv
  Convolutions and more as einsum for PyTorch
  ☆17Updated last year
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆63Updated 2 months ago
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 4 years ago
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆130Updated last month
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆128Updated 10 months ago
• IBM / pylspack
  A lightweight, multithreaded Python package for sketching, column selection, leverage scores and related computations.
  ☆20Updated 2 months ago
• elizabethnewman / hessQuik
  Computing gradients and Hessians of feed-forward networks with GPU acceleration
  ☆19Updated last year
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• google-deepmind / einshape
  ☆104Updated last year
• zalandoresearch / JaxRK
  RKHS feature vectors, operators, and statistical models using JAX for automatic differentiation
  ☆8Updated 4 years ago