nicknytko / nummlLinks
Sparse, differentiable numerics for PyTorch
☆17Updated 5 months ago
Alternatives and similar repositories for numml
Users that are interested in numml are comparing it to the libraries listed below
Sorting:
- A parallel ODE solver for PyTorch☆277Updated last year
- Improved LBFGS and LBFGS-B optimizers in PyTorch.☆67Updated last year
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆275Updated 3 years ago
- PyTorch implementation of the Hessian-free optimizer☆36Updated last year
- ☆16Updated 2 years ago
- A sparse KLU solver for PyTorch.☆66Updated 4 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆123Updated last month
- Newton and Quasi-Newton optimization with PyTorch☆372Updated last week
- Geometric kernels on manifolds, meshes and graphs☆284Updated last month
- Python Algorithms for Randomized Linear Algebra☆57Updated 2 years ago
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆82Updated last year
- Jax implementation of the AdaHessian optimizer☆20Updated 4 years ago
- Sparse matrix tools extending scipy.sparse, but with incompatible licenses☆185Updated last week
- Convolutional Differential Operators for Physics-based Deep Learning Study☆25Updated last year
- L1-regularized least squares with PyTorch☆72Updated 2 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆107Updated 2 years ago
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆69Updated last year
- High-order spline interpolation in PyTorch☆69Updated last year
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆131Updated last week
- Preconditioned Conjugate Gradient in Pytorch☆142Updated last year
- Large-scale, multi-GPU capable, kernel solver☆197Updated 6 months ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆58Updated 6 months ago
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆77Updated 3 months ago
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆34Updated 2 months ago
- GP Sinkhorn Implementation, paper: https://www.mdpi.com/1099-4300/23/9/1134☆23Updated 3 years ago
- An extensible benchmark suite to evaluate data-driven physical simulation☆32Updated 4 years ago
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆68Updated last year
- Fast extremal eigensolvers for PyTorch.☆17Updated 4 years ago
- Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"☆18Updated 2 years ago
- 🚀 A powerful library for efficient training of Neural Fields at scale.☆29Updated 2 years ago