Alternatives and similar repositories for mdmm

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

• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆67Updated 2 years ago
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆130Updated last month
• CW-Huang / CP-Flow
  Convex potential flows
  ☆83Updated 3 years ago
• patrick-kidger / sympytorch
  Turning SymPy expressions into PyTorch modules.
  ☆148Updated last year
• nicola-decao / power_spherical
  Pytorch implementation of the Power Spherical distribution
  ☆74Updated 11 months ago
• google-research / ott
  ☆214Updated 3 years ago
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆34Updated 3 years ago
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆83Updated last year
• massastrello / awesome-implicit-neural-models
  ☆102Updated 4 years ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆63Updated 2 months ago
• patrick-kidger / FasterNeuralDiffEq
  Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)
  ☆87Updated 2 years ago
• FalkonML / falkon
  Large-scale, multi-GPU capable, kernel solver
  ☆190Updated last year
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• johannbrehmer / manifold-flow
  Manifold-learning flows (ℳ-flows)
  ☆230Updated 4 years ago
• toshas / torch-householder
  Efficient Householder Transformation in PyTorch
  ☆66Updated 4 years ago
• toshas / torch_truncnorm
  Truncated Normal Distribution in PyTorch
  ☆84Updated last year
• SymJAX / SymJAX
  Documentation:
  ☆120Updated 2 years ago
• ericjang / nf-jax
  Normalizing Flows in Jax
  ☆107Updated 4 years ago
• CUAI / Neural-Manifold-Ordinary-Differential-Equations
  [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)
  ☆119Updated 2 years ago
• AaltoML / kalman-jax
  Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
  ☆100Updated 2 years ago
• PrincetonLIPS / RandomizedAutomaticDifferentiation
  Experiment code for "Randomized Automatic Differentiation"
  ☆67Updated 4 years ago
• andres-fr / skerch
  Sketched matrix decompositions for PyTorch
  ☆70Updated 2 weeks ago
• sbarratt / torch_cg
  Preconditioned Conjugate Gradient in Pytorch
  ☆140Updated last year
• titu1994 / tf_fourier_features
  Tensorflow 2.0 implementation of Fourier Feature Mapping Networks.
  ☆43Updated 5 years ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆40Updated 4 years ago
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆104Updated last year
• mfinzi / equivariant-MLP
  A library for programmatically generating equivariant layers through constraint solving
  ☆268Updated 2 years ago
• HEmile / storchastic
  Stochastic Automatic Differentiation library for PyTorch.
  ☆204Updated 10 months ago
• conormdurkan / lfi
  Neural likelihood-free methods in PyTorch.
  ☆39Updated 5 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆112Updated 3 years ago