fabiodimarco / torch-levenberg-marquardtLinks
PyTorch implementation of Levenberg-Marquardt training algorithm
☆79Updated 6 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:
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆269Updated 3 years ago
- Newton and Quasi-Newton optimization with PyTorch☆365Updated 4 months ago
- Use scipy.optimize.minimize as a PyTorch Optimizer.☆72Updated last year
- Differentiable signal processing on the sphere for PyTorch☆543Updated this week
- numerical optimization in pytorch☆139Updated last year
- Library to make any existing neural network architecture equivariant☆54Updated 11 months ago
- NumPy arrays, ready for human consumption☆71Updated last week
- Code for our NeurIPS 2022 paper☆369Updated 2 years ago
- RoMa: A lightweight library to deal with 3D rotations in PyTorch.☆604Updated 2 months ago
- Functional N-dimensional convolution in Pytorch, recursively calling convNd until reaching conv3d.☆79Updated 5 months ago
- A tiny library for stochastic dataset caching in PyTorch.☆43Updated last year
- A parallel ODE solver for PyTorch☆267Updated last year
- Universal Notation for Tensor Operations in Python.☆434Updated 6 months ago
- Code repository of the paper "Variational Stochastic Gradient Descent for Deep Neural Networks" published at☆41Updated 5 months ago
- Cubic spline interpolation on multidimensional grids in PyTorch☆33Updated 3 weeks ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆471Updated 11 months ago
- Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning)…☆271Updated this week
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆120Updated 7 months ago
- TUI for the Slurm Workload Manager☆212Updated last month
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆208Updated 2 months ago
- Geometric kernels on manifolds, meshes and graphs☆273Updated last week
- Memory mapped numpy arrays of varying shapes☆302Updated last year
- A Python library for end-to-end learning on surfaces. It implements pre-processing functions that include geodesic algorithms, neural net…☆35Updated last month
- Running Jax in PyTorch Lightning☆112Updated 9 months ago
- Normalizing flows in PyTorch☆410Updated last week
- S2FFT: Differentiable and accelerated spherical transforms☆207Updated last week
- 3D Gaussian Splatting in JAX☆65Updated last year
- Pytorch implementation of same-family gaussian mixture models with guardrails. Features separable parameter optimization and singularity …☆25Updated 4 months ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆287Updated last year
- Run PyTorch in JAX. 🤝☆298Updated last month