Alternatives and similar repositories for veni

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

• NeilGirdhar / tjax
  Tools for JAX
  ☆47Updated last week
• 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
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆54Updated last month
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆49Updated last year
• IBM / pylspack
  A lightweight, multithreaded Python package for sketching, column selection, leverage scores and related computations.
  ☆20Updated last month
• f-dangel / einconv
  Convolutions and more as einsum for PyTorch
  ☆16Updated last year
• elizabethnewman / hessQuik
  Computing gradients and Hessians of feed-forward networks with GPU acceleration
  ☆18Updated last year
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆96Updated last year
• IvanYashchuk / jax-fenics
  Differentiable interface to FEniCS for JAX
  ☆54Updated 4 years ago
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆128Updated 9 months ago
• 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
• luca-heltai / numerical-analysis-2021-2022
  ☆14Updated 2 years ago
• Joshuaalbert / jaxnlds
  Inference on non-linear dynamical systems written in JAX
  ☆11Updated 4 years ago
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆27Updated 3 years ago
• crowsonkb / mdmm-jax
  Gradient-based constrained optimization for JAX
  ☆32Updated 2 years ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 5 months ago
• 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
• cornellius-gp / linear_operator.old
  A LinearOperator implementation for PyTorch
  ☆18Updated 4 years ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆63Updated 2 months ago
• francescocicala / deep-learning-with-pytorch
  Deep learning lectures I am holding for the MSc on Data Science and Scientific Computing
  ☆13Updated 2 years ago
• NeilGirdhar / efax
  Exponential families for JAX
  ☆71Updated this week
• BallisticLA / parla
  Python Algorithms for Randomized Linear Algebra
  ☆54Updated 2 years ago
• juliusberner / deep_kolmogorov
  Numerically Solving Parametric Families of High-Dimensional Kolmogorov Partial Differential Equations via Deep Learning (NeurIPS 2020)
  ☆22Updated 2 years ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆40Updated 4 years ago
• mazy1998 / Solving-Optimization-Problems-with-JAX
  Solving Optimization Problems with JAX, code and PDF
  ☆16Updated 5 years ago
• Romit-Maulik / PyParSVD
  An implementation of the 1. Parallel, 2. Streaming, 3. Randomized SVD using MPI4Py
  ☆62Updated 4 years ago
• juliagusak / neural-ode-norm
  Models and code for the ICLR 2020 workshop paper "Towards Understanding Normalization in Neural ODEs"
  ☆16Updated 5 years ago