Alternatives and similar repositories for gpax

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

• ziatdinovmax / NeuroBayes
  Fully and Partially Bayesian Neural Nets
  ☆69Updated last month
• SergeiVKalinin / UTK-Spring-2023---Automated-Experiment
  Spring 2023 seminar on automated experiment
  ☆23Updated last year
• SergeiVKalinin / MSE_Spring2024
  The materials for the Spring Mathematics in Materials course at the UTK MSE
  ☆49Updated last year
• JaxGaussianProcesses / GPJax
  Gaussian processes in JAX and Flax.
  ☆515Updated 2 weeks ago
• saimani5 / VAE-tutorials
  ☆40Updated last year
• ziatdinovmax / pyroVED
  Invariant representation learning from imaging and spectral data
  ☆51Updated last year
• danielward27 / flowjax
  ☆161Updated 3 weeks ago
• ziatdinovmax / AugmentedGaussianProcess
  Gaussian process augmented with a probabilistic model of expected system's behavior
  ☆15Updated 3 years ago
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆110Updated 2 months ago
• kazewong / flowMC
  Normalizing-flow enhanced sampling package for probabilistic inference in Jax
  ☆239Updated last week
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆119Updated last week
• experimental-design / bofire
  Experimental design and (multi-objective) bayesian optimization.
  ☆288Updated this week
• f0uriest / interpax
  Interpolation and function approximation with JAX
  ☆190Updated 3 weeks ago
• secondmind-labs / trieste
  A Bayesian optimization toolbox built on TensorFlow
  ☆234Updated last week
• dfm / tinygp
  The tiniest of Gaussian Process libraries
  ☆307Updated 3 weeks ago
• e3nn / e3nn-jax
  jax library for E3 Equivariant Neural Networks
  ☆207Updated 4 months ago
• ma921 / BASQ
  (NeurIPS 2022) Fast Bayesian Inference with Batch Bayesian Quadrature via Kernel Recombination
  ☆19Updated last year
• tum-pbs / PhiML
  Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy
  ☆90Updated 2 weeks ago
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆103Updated last year
• google-research / e3x
  E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.
  ☆107Updated last month
• ma921 / SOBER
  Fast Bayesian optimization, quadrature, inference over arbitrary domain with GPU parallel acceleration
  ☆26Updated 3 weeks ago
• florisvb / PyNumDiff
  Methods for numerical differentiation of noisy data in python
  ☆114Updated this week
• PV-Lab / Benchmarking
  Benchmarking
  ☆29Updated 3 years ago
• lbl-camera / fvGP
  A software package for flexible HPC GPs
  ☆16Updated this week
• SergeiVKalinin / MSE_Fall2023
  The materials for the Fall ML in Materials course at the UTK MSE
  ☆85Updated last year
• sissa-data-science / DADApy
  Distance-based Analysis of DAta-manifolds in python
  ☆128Updated last week
• mala-project / mala
  Materials Learning Algorithms. A framework for machine learning materials properties from first-principles data.
  ☆90Updated last week
• qiauil / torchfsm
  TorchFSM: Fourier Spectral Method with PyTorch
  ☆43Updated last week
• aspuru-guzik-group / olympus
  Olympus: a benchmarking framework for noisy optimization and experiment planning
  ☆86Updated 6 months ago
• patrick-kidger / lineax
  Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
  ☆445Updated last week