alexandersoen / pyBregMan

Related projects: ⓘ

• normal-computing / thermox
  Exact OU processes with JAX
  ☆30Updated last month
• AaltoML / PeriodicBNN
  Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)
  ☆16Updated 2 years ago
• justindomke / pangolin
  probabilistic programming focused on fun
  ☆21Updated last week
• Artur-Galstyan / statedict2pytree
  ☆40Updated 2 months ago
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆44Updated 10 months ago
• wiseodd / pub-ready-plots
  Agustinus' very opiniated publication-ready plotting library
  ☆51Updated 2 weeks ago
• pnkraemer / million-dimension-prob-ode-solver-experiments
  ☆10Updated 2 years ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• vr308 / Generalised-Gaussian-Processes
  Fully Bayesian Inference in GPs - Gaussian and Generic Likelihoods
  ☆21Updated 11 months ago
• mancusolab / traceax
  Stochastic trace estimation using JAX
  ☆11Updated 2 months ago
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆40Updated last week
• NeilGirdhar / efax
  Exponential families for JAX
  ☆54Updated last week
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• diffprog / diffprog.github.io
  Website for the book "The Elements of Differentiable Programming".
  ☆13Updated 2 weeks ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆44Updated last year
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆57Updated last month
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆82Updated 9 months ago
• f-dangel / phd-thesis
  Source code for my PhD thesis: Backpropagation Beyond the Gradient
  ☆20Updated last year
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆94Updated 10 months ago
• ml-jku / quam
  Quantification of Uncertainty with Adversarial Models
  ☆27Updated last year
• Iolo-Jones / DiffusionGeometry
  The code for "Diffusion Geometry" (2024).
  ☆22Updated 3 months ago
• cambridge-mlg / sgd-gp
  Public code for running Stochastic Gradient Descent on GPs.
  ☆31Updated 5 months ago
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆43Updated last year
• necludov / action-matching
  Implementation of Action Matching for the Schrödinger equation
  ☆22Updated last year
• cgarciae / einop
  ☆56Updated 2 years ago
• imbirik / LieStationaryKernels
  Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"
  ☆39Updated 10 months ago
• tychovdo / lila
  Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"
  ☆23Updated last year
• uzulim / hades
  Fast singularity detection with kernel
  ☆32Updated 8 months ago
• wesselb / lab
  A generic interface for linear algebra backends
  ☆68Updated 2 months ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆100Updated last month