Alternatives and similar repositories for infrastructure

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

• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆143Updated 4 months ago
• team-approx-bayes / ivon
  IVON optimizer for neural networks based on variational learning.
  ☆72Updated 11 months ago
• wiseodd / pub-ready-plots
  Agustinus' very opiniated publication-ready plotting library
  ☆69Updated 5 months ago
• alisiahkoohi / Langevin-dynamics
  Sampling with gradient-based Markov Chain Monte Carlo approaches
  ☆108Updated last year
• wesselb / neuralprocesses
  A framework for composing Neural Processes in Python
  ☆85Updated 9 months ago
• tristandeleu / jax-dag-gflownet
  Code for "Bayesian Structure Learning with Generative Flow Networks"
  ☆90Updated 3 years ago
• mattclifford1 / IQM-Vis
  ☆25Updated 10 months ago
• NicolasZucchet / minimal-LRU
  Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)
  ☆57Updated last month
• acerbilab / amortized-conditioning-engine
  Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)
  ☆21Updated 3 months ago
• desi-ivanova / idad
  Implicit Deep Adaptive Design (iDAD): Policy-Based Experimental Design without Likelihoods
  ☆20Updated 3 years ago
• andrew-cr / online_var_fil
  Code for our paper: Online Variational Filtering and Parameter Learning
  ☆18Updated 3 years ago
• YannDubs / Neural-Process-Family
  Code for the Neural Processes website and replication of 4 papers on NPs. Pytorch implementation.
  ☆226Updated last year
• locuslab / torchdeq
  Modern Fixed Point Systems using Pytorch
  ☆116Updated last year
• toshas / torch_truncnorm
  Truncated Normal Distribution in PyTorch
  ☆85Updated last year
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆40Updated 2 years ago
• xwinxu / bayeSDE
  Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"
  ☆171Updated 3 years ago
• sail-sg / stde
  Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024
  ☆122Updated 10 months ago
• PetrMokrov / Large-Scale-Wasserstein-Gradient-Flows
  Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)
  ☆37Updated 3 years ago
• HEmile / storchastic
  Stochastic Automatic Differentiation library for PyTorch.
  ☆208Updated last year
• tung-nd / TNP-pytorch
  Official implementation of Transformer Neural Processes
  ☆78Updated 3 years ago
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆130Updated last year
• tranbahien / you-need-a-good-prior
  All You Need is a Good Functional Prior for Bayesian Deep Learning (JMLR 2022)
  ☆20Updated 3 years ago
• google-deepmind / neural_networks_chomsky_hierarchy
  Neural Networks and the Chomsky Hierarchy
  ☆209Updated last year
• jacobjinkelly / easy-neural-ode
  Code for the paper "Learning Differential Equations that are Easy to Solve"
  ☆283Updated 3 years ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆33Updated 9 months ago
• andres-fr / skerch
  Sketched linear operations for PyTorch
  ☆71Updated this week
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆41Updated 4 years ago
• jongharyu / neural-svd
  Official PyTorch implementation of NeuralSVD (ICML 2024)
  ☆20Updated last year
• baumgach / tue-slurm-helloworld
  Instructions and examples to deploy some PyTorch code on slurm using a Singularity Container
  ☆33Updated 2 years ago
• RobertCsordas / onion_representations
  ☆11Updated last year