fabiodimarco / torch-levenberg-marquardtLinks
PyTorch implementation of Levenberg-Marquardt training algorithm
☆86Updated this week
Alternatives and similar repositories for torch-levenberg-marquardt
Users that are interested in torch-levenberg-marquardt are comparing it to the libraries listed below
Sorting:
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆272Updated 3 years ago
- Differentiable signal processing on the sphere for PyTorch☆601Updated last week
- Newton and Quasi-Newton optimization with PyTorch☆368Updated 6 months ago
- Use scipy.optimize.minimize as a PyTorch Optimizer.☆73Updated last year
- NumPy arrays, ready for human consumption☆72Updated last week
- RoMa: A lightweight library to deal with 3D rotations in PyTorch.☆618Updated 4 months ago
- numerical optimization in pytorch☆139Updated last year
- Code repository of the paper "Variational Stochastic Gradient Descent for Deep Neural Networks" published at☆41Updated 6 months ago
- Library to make any existing neural network architecture equivariant☆55Updated last year
- Code for our NeurIPS 2022 paper☆370Updated 2 years ago
- 3D Gaussian Splatting in JAX☆65Updated last year
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆481Updated last year
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆210Updated 3 months ago
- Geometric kernels on manifolds, meshes and graphs☆279Updated last week
- A parallel ODE solver for PyTorch☆271Updated last year
- Supplemental learning materials for "Fourier Feature Networks and Neural Volume Rendering"☆179Updated 2 years ago
- A tiny library for stochastic dataset caching in PyTorch.☆44Updated last year
- S2FFT: Differentiable and accelerated spherical transforms☆208Updated this week
- Cubic spline interpolation on multidimensional grids in PyTorch☆34Updated this week
- GPU/TPU accelerated nonlinear least-squares curve fitting using JAX☆58Updated 2 years ago
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆25Updated last week
- TUI for the Slurm Workload Manager☆230Updated 3 months ago
- Run PyTorch in JAX. 🤝☆307Updated last month
- ☆128Updated 11 months ago
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆49Updated 6 months ago
- Exact method for visualizing partitions of a Deep Neural Network, CVPR 2023 Highlight☆110Updated 9 months ago
- Universal Notation for Tensor Operations in Python.☆449Updated 7 months ago
- Memory mapped numpy arrays of varying shapes☆304Updated last year
- 3d without friction (Torch, TF, Jax, Numpy)☆162Updated last year
- Pairwise Metrics for PyTorch☆20Updated 6 months ago