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
  ☆67Updated 2 weeks ago
• SergeiVKalinin / UTK-Spring-2023---Automated-Experiment
  Spring 2023 seminar on automated experiment
  ☆23Updated last year
• JaxGaussianProcesses / GPJax
  Gaussian processes in JAX and Flax.
  ☆504Updated last week
• secondmind-labs / trieste
  A Bayesian optimization toolbox built on TensorFlow
  ☆229Updated 3 weeks ago
• SergeiVKalinin / MSE_Spring2024
  The materials for the Spring Mathematics in Materials course at the UTK MSE
  ☆49Updated 11 months ago
• saimani5 / VAE-tutorials
  ☆40Updated last year
• danielward27 / flowjax
  ☆155Updated last month
• ziatdinovmax / pyroVED
  Invariant representation learning from imaging and spectral data
  ☆50Updated last year
• tum-pbs / PhiML
  Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy
  ☆84Updated last week
• secondmind-labs / GPflux
  Deep GPs built on top of TensorFlow/Keras and GPflow
  ☆125Updated 6 months ago
• ziatdinovmax / AugmentedGaussianProcess
  Gaussian process augmented with a probabilistic model of expected system's behavior
  ☆15Updated 3 years ago
• kazewong / flowMC
  Normalizing-flow enhanced sampling package for probabilistic inference in Jax
  ☆230Updated this week
• ma921 / BASQ
  (NeurIPS 2022) Fast Bayesian Inference with Batch Bayesian Quadrature via Kernel Recombination
  ☆19Updated last year
• experimental-design / bofire
  Experimental design and (multi-objective) bayesian optimization.
  ☆274Updated this week
• google-research / e3x
  E3x is a JAX library for constructing efficient E(3)-equivariant deep learning architectures built on top of Flax.
  ☆103Updated 2 weeks ago
• e3nn / e3nn-jax
  jax library for E3 Equivariant Neural Networks
  ☆204Updated 3 months ago
• f0uriest / interpax
  Interpolation and function approximation with JAX
  ☆172Updated 2 weeks ago
• ma921 / SOBER
  Fast Bayesian optimization, quadrature, inference over arbitrary domain with GPU parallel acceleration
  ☆26Updated last month
• lbl-camera / fvGP
  A software package for flexible HPC GPs
  ☆16Updated last week
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆103Updated last year
• PV-Lab / Benchmarking
  Benchmarking
  ☆28Updated 3 years ago
• dfm / tinygp
  The tiniest of Gaussian Process libraries
  ☆306Updated last week
• ziatdinovmax / GPim
  Gaussian processes and Bayesian optimization for images and hyperspectral data
  ☆57Updated last year
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆116Updated 3 months ago
• SergeiVKalinin / ML-ElectronMicroscopy-2023
  ☆121Updated last year
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆408Updated this week
• patrick-kidger / lineax
  Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
  ☆433Updated 3 weeks ago
• google-deepmind / flows_for_atomic_solids
  ☆49Updated 2 years ago
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆107Updated last month
• scifm / summer-school-2024
  ☆37Updated 8 months ago