nicknytko / numml
☆14Updated 6 months ago
Alternatives and similar repositories for numml:
Users that are interested in numml are comparing it to the libraries listed below
- Convex potential flows☆81Updated 3 years ago
- Code for 'Periodic Activation Functions Induce Stationarity' (NeurIPS 2021)☆18Updated 3 years ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆42Updated last year
- L1-regularized least squares with PyTorch☆64Updated last year
- PyTorch implementation of the Hessian-free optimizer☆33Updated 7 months ago
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆27Updated 8 months ago
- ☆16Updated last year
- Neural Walk-on-Spheres☆11Updated 6 months ago
- Equivariant Mesh Attention Networks☆21Updated 2 years ago
- Learning incomplete factorization preconditioners using graph neural networks☆11Updated last week
- An extensible benchmark suite to evaluate data-driven physical simulation☆24Updated 3 years ago
- Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"☆12Updated last year
- Normalizing flows for neural importance sampling☆37Updated 7 months ago
- ☆63Updated last year
- [ICML 2022] Learning Efficient and Robust Ordinary Differential \\ Equations via Invertible Neural Networks☆9Updated last year
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- A sparse KLU solver for PyTorch.☆62Updated 3 years ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆69Updated 8 months ago
- The Modified Differential Multiplier Method (MDMM) for PyTorch☆55Updated 3 years ago
- Computing gradients and Hessians of feed-forward networks with GPU acceleration☆18Updated 11 months ago
- A tool for generating PDEs ground truth datasets from ARCSim, FEniCS and SU2☆35Updated 3 years ago
- Learning Algebraic Multigrid Using Graph Neural Networks☆20Updated 4 years ago
- Riemannian Convex Potential Maps☆67Updated last year
- ☆16Updated 2 years ago
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆73Updated 3 weeks ago
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆72Updated 5 months ago
- Code for Gaussian Score Matching Variational Inference☆28Updated 3 months ago
- Scale-invariant Learning by Physics Inversion (NeurIPS 2022)☆10Updated 2 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆43Updated last week
- A C++/Python library for incomplete LU factorizations based on Jan Mayer's ILU++☆28Updated 3 years ago