Alternatives and similar repositories for pgm-jax

Users that are interested in pgm-jax are comparing it to the libraries listed below

• probml / probml-utils
  Utilities for probabilistic ML
  ☆36Updated 2 years ago
• probml / JSL
  Jax SSM Library
  ☆48Updated 3 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆47Updated 5 years ago
• vicariousinc / PGMax
  Loopy belief propagation for factor graphs on discrete variables, in JAX!
  ☆65Updated last year
• probml / bandits
  Bayesian Bandits
  ☆68Updated 2 years ago
• harvardnlp / strux
  ☆18Updated 3 years ago
• srush / jax-lda
  ☆30Updated 4 years ago
• ethanluoyc / dhmc-jax
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆42Updated 5 years ago
• pyro-ppl / sandbox
  Pyro models and misc examples.
  ☆19Updated 4 years ago
• google / deluca
  Performant, differentiable reinforcement learning
  ☆125Updated 6 months ago
• ivaleraM / DataTypes
  Automatic Discovery of the Statistical Types of Variables in a Dataset
  ☆23Updated 7 years ago
• google-research / sloe-logistic
  ☆29Updated last year
• mblondel / projection-losses
  Python implementation of projection losses.
  ☆27Updated 6 years ago
• AaltoML / kalman-jax
  Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
  ☆103Updated 2 years ago
• mgorinova / autoreparam
  Automatic Reparameterisation of Probabilistic Programs
  ☆36Updated 5 years ago
• distillpub / post--visual-exploration-gaussian-processes
  A Visual Exploration of Gaussian Processes
  ☆107Updated 3 years ago
• SymJAX / SymJAX
  Documentation:
  ☆127Updated 2 years ago
• facebookresearch / pplbench
  Evaluation Framework for Probabilistic Programming Languages
  ☆104Updated 2 years ago
• oxcsml / ML_bazaar
  OxCSML research group reading groups and meetings at the Department of Statistics, University of Oxford.
  ☆96Updated 4 years ago
• google-deepmind / einshape
  ☆109Updated last year
• plai-group / covid
  Modelling epidemiological dynamics and performing inference in these models
  ☆27Updated 4 years ago
• pyro-ppl / brmp
  Bayesian Regression Models in Pyro
  ☆74Updated last year
• hanyas / mimo
  A toolbox for inference of mixture models
  ☆16Updated 2 years ago
• HIPS / pgmult
  Dependent multinomials made easy: stick-breaking with the Pólya-gamma augmentation
  ☆61Updated 5 years ago
• google-deepmind / neural_testbed
  ☆195Updated last month
• google-research / autoconj
  Recognizing and exploiting conjugacy without a domain-specific language
  ☆36Updated 6 years ago
• rgiordan / CovariancesRobustnessVBPaper
  ☆12Updated 2 years ago
• mcleonard / xaby
  Functional machine learning for fun
  ☆85Updated 4 years ago
• cgarciae / einop
  ☆60Updated 3 years ago
• pierreablin / autoptim
  Automatic differentiation + optimization
  ☆101Updated 6 years ago