Alternatives and similar repositories for infrastructure

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

• alisiahkoohi / Langevin-dynamics
  Sampling with gradient-based Markov Chain Monte Carlo approaches
  ☆108Updated last year
• andrew-cr / online_var_fil
  Code for our paper: Online Variational Filtering and Parameter Learning
  ☆20Updated 4 years ago
• wiseodd / pub-ready-plots
  Agustinus' very opiniated publication-ready plotting library
  ☆69Updated 7 months ago
• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆23Updated 3 years ago
• NicolasZucchet / minimal-LRU
  Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)
  ☆61Updated 3 months ago
• tristandeleu / jax-dag-gflownet
  Code for "Bayesian Structure Learning with Generative Flow Networks"
  ☆94Updated 3 years ago
• toshas / torch_truncnorm
  Truncated Normal Distribution in PyTorch
  ☆87Updated last year
• neuraloperator / GANO
  ☆35Updated 2 years ago
• YannDubs / Neural-Process-Family
  Code for the Neural Processes website and replication of 4 papers on NPs. Pytorch implementation.
  ☆227Updated last year
• DSL-Lab / GRBM
  Gaussian-Bernoulli Restricted Boltzmann Machines
  ☆106Updated 3 years ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆42Updated 5 years ago
• team-approx-bayes / ivon
  IVON optimizer for neural networks based on variational learning.
  ☆75Updated last year
• iffsid / DReG-PyTorch
  A PyTorch re-implementation of "Doubly Reparameterized Gradient Estimators for Monte Carlo Objectives"
  ☆18Updated 6 years ago
• mattclifford1 / IQM-Vis
  ☆25Updated 2 weeks ago
• probabilisticai / probai-2022
  Materials of the Nordic Probabilistic AI School 2022.
  ☆181Updated 3 years ago
• tung-nd / TNP-pytorch
  Official implementation of Transformer Neural Processes
  ☆78Updated 3 years ago
• frgsimpson / kitt
  Kernel Identification Through Transformers
  ☆14Updated 2 years ago
• jacobjinkelly / easy-neural-ode
  Code for the paper "Learning Differential Equations that are Easy to Solve"
  ☆286Updated 4 years ago
• HEmile / storchastic
  Stochastic Automatic Differentiation library for PyTorch.
  ☆208Updated last year
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆130Updated last year
• wesselb / neuralprocesses
  A framework for composing Neural Processes in Python
  ☆89Updated 11 months ago
• google-deepmind / neural_networks_chomsky_hierarchy
  Neural Networks and the Chomsky Hierarchy
  ☆212Updated last year
• acerbilab / amortized-conditioning-engine
  Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)
  ☆21Updated 5 months ago
• MachineLearningLifeScience / stochman
  Algorithms for computations on random manifolds made easier
  ☆94Updated 2 years ago
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆41Updated 3 years ago
• april-tools / squared-npcs
  Squared Non-monotonic Probabilistic Circuits
  ☆23Updated 11 months ago
• andres-fr / skerch
  Sketched linear operations for PyTorch
  ☆99Updated last month
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆35Updated 11 months ago
• bb515 / diffusionjax
  diffusionjax is a simple and accessible diffusion models package in JAX
  ☆48Updated 10 months ago
• cagatayyildiz / ODE2VAE
  ODE2VAE: Deep generative second order ODEs with Bayesian neural networks
  ☆130Updated last year