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 2 years 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
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 3 years ago
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆28Updated 4 years ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 11 months ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆58Updated 4 months ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆49Updated last month
• j-towns / scanagram
  Tidy autoregressive inference in JAX
  ☆15Updated 3 months ago
• cgarciae / einop
  ☆60Updated 3 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
• 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.
  ☆38Updated 3 years ago
• crowsonkb / mdmm-jax
  Gradient-based constrained optimization for JAX
  ☆37Updated 3 years ago
• elizabethnewman / hessQuik
  Computing gradients and Hessians of feed-forward networks with GPU acceleration
  ☆20Updated last year
• IBM / pylspack
  A lightweight, multithreaded Python package for sketching, column selection, leverage scores and related computations.
  ☆20Updated 2 months ago
• google-deepmind / einshape
  ☆107Updated last year
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆131Updated last year
• gehring / fax
  ☆80Updated 4 years ago
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆146Updated 3 weeks ago
• cgarciae / dyn_plot
  ☆13Updated last year
• juliagusak / neural-ode-metasolver
  Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561
  ☆25Updated 4 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆45Updated 5 years ago
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆30Updated 5 years ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated 2 years ago
• francescocicala / deep-learning-with-pytorch
  Deep learning lectures I am holding for the MSc on Data Science and Scientific Computing
  ☆14Updated 3 years ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• facebookresearch / semi-discrete-flow
  code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"
  ☆27Updated 3 years ago
• formll / dog
  DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule
  ☆63Updated 2 years ago
• JeanKossaifi / zencfg
  A Zen approach to configuring your Python project
  ☆15Updated 3 months ago