Alternatives and similar repositories for veni

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

• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆53Updated last year
• NeilGirdhar / tjax
  Tools for JAX
  ☆49Updated 3 weeks ago
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated 2 years ago
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆143Updated 4 months ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• JeanKossaifi / zencfg
  A Zen approach to configuring your Python project
  ☆15Updated last month
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆28Updated 4 years ago
• IBM / pylspack
  A lightweight, multithreaded Python package for sketching, column selection, leverage scores and related computations.
  ☆20Updated 3 weeks ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆58Updated 2 months ago
• google-deepmind / einshape
  ☆106Updated last year
• google-deepmind / PGMax
  Loopy belief propagation for factor graphs on discrete variables in JAX
  ☆157Updated 11 months ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆45Updated 4 years ago
• crowsonkb / mdmm-jax
  Gradient-based constrained optimization for JAX
  ☆35Updated 3 years ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆107Updated 3 months ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆67Updated 5 months 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
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆108Updated last year
• andres-fr / skerch
  Sketched linear operations for PyTorch
  ☆71Updated this week
• cambridge-mlg / EinsumNetworks
  ☆54Updated 2 years ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 9 months ago
• moskomule / hypergrad
  Simple and extensible hypergradient for PyTorch
  ☆18Updated 2 years ago
• j-towns / scanagram
  Tidy autoregressive inference in JAX
  ☆14Updated last month
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆128Updated 3 weeks ago
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆30Updated 5 years ago
• toshas / torch-householder
  Efficient Householder Transformation in PyTorch
  ☆66Updated 4 years ago
• google / tree-math
  Mathematical operations for JAX pytrees
  ☆200Updated 10 months ago
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆50Updated 4 years ago
• google-deepmind / tf2jax
  ☆115Updated last month
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆130Updated last year