nestordemeure / AdaHessianJax

Related projects ⓘ

Alternatives and complementary repositories for AdaHessianJax

• davda54 / ada-hessian
  Easy-to-use AdaHessian optimizer (PyTorch)
  ☆77Updated 4 years ago
• AlexanderMath / fasth
  Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.
  ☆69Updated 3 months ago
• patrick-kidger / FasterNeuralDiffEq
  Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)
  ☆86Updated 2 years ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …
  ☆110Updated 3 weeks ago
• HazyResearch / structured-nets
  Structured matrices for compressing neural networks
  ☆67Updated last year
• CW-Huang / CP-Flow
  Convex potential flows
  ☆78Updated 2 years ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆37Updated 4 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆104Updated 2 years ago
• juntang-zhuang / TorchDiffEqPack
  ☆45Updated last year
• danieljtait / jax_xla_adventures
  Dive into Jax, Flax, XLA and C++
  ☆31Updated 4 years ago
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆185Updated 5 months ago
• g-benton / loss-surface-simplexes
  ☆97Updated 2 years ago
• google-deepmind / annealed_flow_transport
  ☆49Updated last year
• juntang-zhuang / torch_ACA
  repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE
  ☆55Updated 3 years ago
• necludov / action-matching
  Implementation of Action Matching for the Schrödinger equation
  ☆22Updated last year
• locuslab / monotone_op_net
  Monotone operator equilibrium networks
  ☆51Updated 4 years ago
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆82Updated 11 months ago
• gehring / fax
  ☆78Updated 3 years ago
• mbilos / stribor
  Library for normalizing flows and neural flows.
  ☆24Updated 2 years ago
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆95Updated last year
• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆68Updated last year
• juliagusak / neural-ode-metasolver
  Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561
  ☆25Updated 3 years ago
• flaport / torch_eunn
  A Pytorch implementation of an efficient unitary neural network (https://arxiv.org/abs/1612.05231)
  ☆32Updated 4 years ago
• tk-rusch / unicornn
  Official code for UnICORNN (ICML 2021)
  ☆27Updated 3 years ago
• plai-group / tvo
  Code for the Thermodynamic Variational Objective
  ☆26Updated 2 years ago
• tensorly / torch
  TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch
  ☆76Updated 5 months ago
• conormdurkan / lfi
  Neural likelihood-free methods in PyTorch.
  ☆39Updated 4 years ago
• ml-research / pau
  Padé Activation Units: End-to-end Learning of Activation Functions in Deep Neural Network
  ☆64Updated 3 years ago
• dimitra-maoutsa / odes_for_sdes
  Deterministic particle dynamics for simulating Fokker-Planck probability flows
  ☆24Updated last year
• afqueiruga / StatefulOdeNets
  Refining continuous-in-depth neural networks
  ☆39Updated 2 years ago