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.
  ☆53Updated last month
• Romit-Maulik / PyParSVD
  An implementation of the 1. Parallel, 2. Streaming, 3. Randomized SVD using MPI4Py
  ☆61Updated 4 years ago
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆49Updated last year
• luca-heltai / numerical-analysis-2021-2022
  ☆14Updated 2 years ago
• yhtang / FunFact
  Tensor decomposition with arbitrary expressions: inner, outer, elementwise operators; nonlinear transformations; and more.
  ☆58Updated 2 years ago
• 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
• NeilGirdhar / tjax
  Tools for JAX
  ☆47Updated last week
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆128Updated 8 months ago
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆96Updated last year
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆22Updated 5 months ago
• pnkraemer / probfindiff
  Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…
  ☆11Updated 2 years ago
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆27Updated 3 years ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• elizabethnewman / hessQuik
  Computing gradients and Hessians of feed-forward networks with GPU acceleration
  ☆18Updated last year
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 4 months ago
• mfinzi / constrained-hamiltonian-neural-networks
  ☆107Updated 4 years ago
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆38Updated 2 years ago
• crowsonkb / mdmm-jax
  Gradient-based constrained optimization for JAX
  ☆32Updated 2 years ago
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆86Updated 2 months ago
• MJHutchinson / ExtrinsicGaugeIndependentVectorGPs
  A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes
  ☆25Updated 3 years ago
• ASEM000 / serket
  The ✨Magical✨ JAX ML Library.
  ☆16Updated 4 months ago
• BallisticLA / parla
  Python Algorithms for Randomized Linear Algebra
  ☆54Updated 2 years ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆124Updated this week
• 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
• ASEM000 / kernex
  Stencil computations in JAX
  ☆71Updated last year
• liuyao12 / ConvNets-PDE-perspective
  an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets
  ☆22Updated last year
• stfc-sciml / sciml-bench
  SciML Benchmarking Suite for AI for Science
  ☆41Updated 10 months ago
• sail-sg / autofd
  Automatic Functional Differentiation in JAX
  ☆72Updated 2 months ago
• f-dangel / einconv
  Convolutions and more as einsum for PyTorch
  ☆16Updated last year