tensorly / torch
TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch
☆78Updated 10 months ago
Alternatives and similar repositories for torch:
Users that are interested in torch are comparing it to the libraries listed below
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆74Updated 8 months ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆87Updated 2 years ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆37Updated 2 years ago
- Riemannian Optimization Using JAX☆48Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆107Updated last month
- Convex potential flows☆83Updated 3 years ago
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆169Updated 3 years ago
- ☆101Updated 4 years ago
- This repository implements and evaluates convolutional networks on the Möbius strip as toy model instantiations of Coordinate Independent…☆72Updated last year
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- ☆47Updated 5 years ago
- ☆46Updated last year
- Turning SymPy expressions into PyTorch modules.☆147Updated last year
- The simplest pytorch implement (100 lines) of "Neural Ordinary Differential Equations" @ NeurIPS 2018 Best Paper.☆77Updated 4 years ago
- [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)☆117Updated last year
- Second-Order Neural ODE Optimizer, NeurIPS 2021 spotlight☆52Updated 3 years ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Lie Algebra Convolutional Network implementation☆43Updated 3 years ago
- Matrix-free linear algebra in JAX.☆116Updated 3 months ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆115Updated last week
- Computing gradients and Hessians of feed-forward networks with GPU acceleration☆18Updated last year
- Pytorch implementation of KFAC - this is a port of https://github.com/tensorflow/kfac/☆23Updated 10 months ago
- Tensor Network Learning with PyTorch☆295Updated 10 months ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆106Updated 4 years ago
- Algorithms for computations on random manifolds made easier☆90Updated last year
- Experiments for Neural Flows paper