tensorly / torch

Related projects ⓘ

Alternatives and complementary repositories for torch

• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆45Updated last year
• 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
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆94Updated 2 months ago
• kazukiosawa / asdl
  ASDL: Automatic Second-order Differentiation Library for PyTorch
  ☆179Updated 3 months ago
• xwinxu / bayeSDE
  Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"
  ☆170Updated 2 years ago
• patrick-kidger / FasterNeuralDiffEq
  Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)
  ☆86Updated 2 years ago
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆107Updated this week
• CUAI / Neural-Manifold-Ordinary-Differential-Equations
  [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)
  ☆115Updated last year
• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆68Updated last year
• ghliu / snopt
  Second-Order Neural ODE Optimizer, NeurIPS 2021 spotlight
  ☆52Updated 3 years ago
• juntang-zhuang / TorchDiffEqPack
  ☆45Updated last year
• f-dangel / curvlinops
  scipy linear operators for the Hessian, Fisher/GGN, and more in PyTorch
  ☆18Updated 2 weeks ago
• google-deepmind / kfac-jax
  Second Order Optimization and Curvature Estimation with K-FAC in JAX.
  ☆249Updated this week
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆35Updated last year
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆82Updated 11 months ago
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆106Updated 2 months ago
• noegroup / stochastic_normalizing_flows
  Stochastic Normalizing Flows
  ☆73Updated 2 years ago
• VectorInstitute / vbll
  Simple (and cheap!) neural network uncertainty estimation
  ☆55Updated 2 months ago
• brandstetter-johannes / LPSDA
  Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"
  ☆48Updated last year
• massastrello / awesome-implicit-neural-models
  ☆98Updated 3 years ago
• alisiahkoohi / Langevin-dynamics
  Sampling with gradient-based Markov Chain Monte Carlo approaches
  ☆89Updated 7 months ago
• DiffEqML / diffeqml-research
  This repository contains code released by DiffEqML Research
  ☆85Updated 2 years ago
• CW-Huang / CP-Flow
  Convex potential flows
  ☆78Updated 3 years ago
• necludov / jam
  Implementation of Action Matching
  ☆36Updated last year
• f-dangel / vivit
  [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…
  ☆17Updated last year
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …
  ☆127Updated last month
• zoq / Awesome-Optimizer
  Collect optimizer related papers, data, repositories
  ☆82Updated this week
• mauriceweiler / MobiusCNNs
  This repository implements and evaluates convolutional networks on the Möbius strip as toy model instantiations of Coordinate Independent…
  ☆73Updated 11 months ago
• gpauloski / kfac-pytorch
  Distributed K-FAC Preconditioner for PyTorch
  ☆80Updated this week
• nimadehmamy / L-conv-code
  Lie Algebra Convolutional Network implementation
  ☆42Updated 2 years ago