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
  ☆52Updated last year
• NeilGirdhar / tjax
  Tools for JAX
  ☆49Updated this week
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆57Updated last month
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆84Updated last year
• crowsonkb / mdmm-jax
  Gradient-based constrained optimization for JAX
  ☆35Updated 3 years ago
• gehring / fax
  ☆80Updated 4 years ago
• google-research / gpax
  ☆30Updated last year
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆140Updated 3 months ago
• EEA-sensors / sequential-parallelization-examples
  This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.
  ☆33Updated 2 years ago
• IBM / pylspack
  A lightweight, multithreaded Python package for sketching, column selection, leverage scores and related computations.
  ☆20Updated 4 months ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 8 months ago
• NeilGirdhar / efax
  Exponential families for JAX
  ☆73Updated last week
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆28Updated 3 years ago
• google / tree-math
  Mathematical operations for JAX pytrees
  ☆200Updated 9 months ago
• ASEM000 / kernex
  Stencil computations in JAX
  ☆71Updated last year
• gbaydin / difftorch
  A differentiation API for PyTorch
  ☆30Updated 5 years ago
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆130Updated last year
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• ethanluoyc / dhmc-jax
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆42Updated 5 years ago
• toshas / torch-householder
  Efficient Householder Transformation in PyTorch
  ☆66Updated 4 years ago
• google-deepmind / PGMax
  Loopy belief propagation for factor graphs on discrete variables in JAX
  ☆156Updated 11 months ago
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆50Updated 4 years ago
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆105Updated last year
• openopt / chop
  CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.
  ☆78Updated last year
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆40Updated last year
• SymJAX / SymJAX
  Documentation:
  ☆122Updated 2 years ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆45Updated 4 years ago
• google-deepmind / einshape
  ☆105Updated last year
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆106Updated 2 months ago