fabiodimarco / torch-levenberg-marquardtLinks
PyTorch implementation of Levenberg-Marquardt training algorithm
☆77Updated 5 months ago
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:
- Newton and Quasi-Newton optimization with PyTorch☆363Updated 3 months ago
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆264Updated 2 years ago
- High-order spline interpolation in PyTorch☆68Updated last year
- Differentiable signal processing on the sphere for PyTorch☆529Updated last week
- numerical optimization in pytorch☆139Updated last year
- Use scipy.optimize.minimize as a PyTorch Optimizer.☆72Updated last year
- NumPy arrays, ready for human consumption☆71Updated 3 months ago
- Code repository of the paper "Variational Stochastic Gradient Descent for Deep Neural Networks" published at☆41Updated 4 months ago
- Code for our NeurIPS 2022 paper☆369Updated 2 years ago
- Library to make any existing neural network architecture equivariant☆54Updated 10 months ago
- 3D Gaussian Splatting in JAX☆63Updated last year
- Cubic spline interpolation on multidimensional grids in PyTorch☆33Updated 2 weeks ago
- Functional N-dimensional convolution in Pytorch, recursively calling convNd until reaching conv3d.☆79Updated 4 months ago
- RoMa: A lightweight library to deal with 3D rotations in PyTorch.☆601Updated last month
- A tiny library for stochastic dataset caching in PyTorch.☆43Updated last year
- Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning)…☆269Updated this week
- Supplemental learning materials for "Fourier Feature Networks and Neural Volume Rendering"☆179Updated 2 years ago
- TUI for the Slurm Workload Manager☆208Updated 3 weeks ago
- A parallel ODE solver for PyTorch☆265Updated 11 months ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆207Updated last month
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆414Updated 5 months ago
- ☆158Updated 2 years ago
- Geometric kernels on manifolds, meshes and graphs☆271Updated 2 months ago
- High order and sparse layers in pytorch. Lagrange Polynomial, Piecewise Lagrange Polynomial, Piecewise Discontinuous Lagrange Polynomial…☆45Updated last year
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆25Updated 7 months ago
- Efficiently Composable Data Augmentation on the GPU with Jax☆39Updated 4 months ago
- Run PyTorch in JAX. 🤝☆290Updated 2 weeks ago
- GPU/TPU accelerated nonlinear least-squares curve fitting using JAX☆58Updated 2 years ago
- Integral Neural Networks in PyTorch☆125Updated 9 months ago
- Exact method for visualizing partitions of a Deep Neural Network, CVPR 2023 Highlight☆109Updated 7 months ago