Alternatives and similar repositories for mlxp:

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

• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆170Updated 3 months ago
• google-deepmind / conformal_training
  This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…
  ☆129Updated 2 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆43Updated 4 years ago
• ctlllll / SGConv
  ☆164Updated 2 years ago
• aradha / recursive_feature_machines
  ☆43Updated this week
• microsoft / bayesianize
  Bayesianize: A Bayesian neural network wrapper in pytorch
  ☆88Updated 10 months ago
• team-approx-bayes / ivon
  IVON optimizer for neural networks based on variational learning.
  ☆60Updated 4 months ago
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆122Updated 11 months ago
• shikaiqiu / compute-better-spent
  ☆52Updated 5 months ago
• openopt / chop
  CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.
  ☆77Updated last year
• captain-pool / hydra-wandb-sweeper
  WandB sweeps integration with Hydra sweeper
  ☆47Updated last year
• OATML / RHO-Loss
  ☆201Updated 2 years ago
• SamsungLabs / BayesDLL
  ☆145Updated last year
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆321Updated last year
• HEmile / storchastic
  Stochastic Automatic Differentiation library for PyTorch.
  ☆197Updated 6 months ago
• google / spectral-density
  Hessian spectral density estimation in TF and Jax
  ☆122Updated 4 years ago
• cgarciae / einop
  ☆60Updated 3 years ago
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆112Updated this week
• google-deepmind / synjax
  ☆245Updated 5 months ago
• google-deepmind / einshape
  ☆101Updated 8 months ago
• g-benton / loss-surface-simplexes
  ☆99Updated 3 years ago
• f-dangel / sirfshampoo
  [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)
  ☆14Updated 4 months ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆103Updated 8 months ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• f-dangel / curvlinops
  PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)
  ☆34Updated 2 weeks ago
• apple / ml-calibration
  relplot: Utilities for measuring calibration and plotting reliability diagrams
  ☆141Updated 9 months ago
• google-deepmind / tf2jax
  ☆112Updated last month
• facebookincubator / flowtorch
  This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…
  ☆305Updated last week
• stanislavfort / dissect-git-re-basin
  Replicating and dissecting the git-re-basin project in one-click-replication Colabs
  ☆36Updated 2 years ago
• y0ast / slurm-for-ml
  A Machine Learning workflow for Slurm.
  ☆149Updated 4 years ago