Alternatives and similar repositories for mlxp:

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

• SamsungLabs / BayesDLL
  ☆143Updated last year
• google-deepmind / conformal_training
  This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…
  ☆128Updated 2 years ago
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆109Updated 8 months ago
• openopt / chop
  CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.
  ☆77Updated 10 months ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆148Updated last month
• probabilisticai / probai-2023
  Materials of the Nordic Probabilistic AI School 2023.
  ☆85Updated last year
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆110Updated this week
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆42Updated 4 years ago
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆14Updated 2 months ago
• NeilGirdhar / tjax
  Tools for JAX
  ☆44Updated this week
• stanislavfort / dissect-git-re-basin
  Replicating and dissecting the git-re-basin project in one-click-replication Colabs
  ☆36Updated 2 years ago
• microsoft / bayesianize
  Bayesianize: A Bayesian neural network wrapper in pytorch
  ☆88Updated 8 months ago
• OATML / RHO-Loss
  ☆198Updated 2 years ago
• google / spectral-density
  Hessian spectral density estimation in TF and Jax
  ☆120Updated 4 years ago
• google-deepmind / synjax
  ☆243Updated 3 months ago
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆39Updated last year
• AaltoML / kalman-jax
  Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
  ☆98Updated last year
• shikaiqiu / compute-better-spent
  ☆50Updated 3 months ago
• LaihoE / did-it-spill
  Check if you have training samples in your test set
  ☆64Updated 2 years ago
• davda54 / ada-hessian
  Easy-to-use AdaHessian optimizer (PyTorch)
  ☆77Updated 4 years ago
• blackjax-devs / sampling-book
  Tutorials and sampling algorithm comparisons
  ☆70Updated this week
• team-approx-bayes / ivon
  IVON optimizer for neural networks based on variational learning.
  ☆57Updated 2 months ago
• wiseodd / pub-ready-plots
  Agustinus' very opiniated publication-ready plotting library
  ☆61Updated 4 months ago
• VectorInstitute / vbll
  Simple (and cheap!) neural network uncertainty estimation
  ☆59Updated this week
• runame / laplace-redux
  Laplace Redux -- Effortless Bayesian Deep Learning
  ☆41Updated last year
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆47Updated 3 years ago
• rbardenet / bml-course
  Public repo for course material on Bayesian machine learning at ENS Paris-Saclay and Univ Lille
  ☆81Updated this week
• mathurinm / example_package
  ☆13Updated last year
• apple / ml-calibration
  relplot: Utilities for measuring calibration and plotting reliability diagrams
  ☆136Updated 7 months ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆57Updated last year