fabiodimarco / torch-levenberg-marquardt
PyTorch implementation of Levenberg-Marquardt training algorithm
☆57Updated last month
Alternatives and similar repositories for torch-levenberg-marquardt:
Users that are interested in torch-levenberg-marquardt are comparing it to the libraries listed below
- Newton and Quasi-Newton optimization with PyTorch☆342Updated last year
- Library to make any existing neural network architecture equivariant☆52Updated 5 months ago
- numerical optimization in pytorch☆136Updated 11 months ago
- GPU/TPU accelerated nonlinear least-squares curve fitting using JAX☆56Updated last year
- NumPy arrays, ready for human consumption☆69Updated 9 months ago
- Use scipy.optimize.minimize as a PyTorch Optimizer.☆72Updated 9 months ago
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆251Updated 2 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆107Updated last month
- Run PyTorch in JAX. 🤝☆236Updated 2 months ago
- Running Jax in PyTorch Lightning☆94Updated 4 months ago
- Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"☆47Updated 8 months ago
- Code for our NeurIPS 2022 paper☆367Updated 2 years ago
- ☆152Updated last year
- A Python package of computer vision models for the Equinox ecosystem.☆103Updated 9 months ago
- 3D Gaussian Splatting in JAX☆57Updated 10 months ago
- Differentiable signal processing on the sphere for PyTorch☆451Updated last month
- Memory mapped numpy arrays of varying shapes☆297Updated 10 months ago
- NumPy-style histograms in PyTorch☆54Updated last year
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆51Updated 3 weeks ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆204Updated 5 months ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆115Updated 3 weeks ago
- Constrained optimization toolkit for PyTorch☆673Updated 3 years ago
- A functional training loops library for JAX☆86Updated last year
- Code for "Manifold Diffusion Geometry: Curvature, Tangent Spaces, and Dimension"☆43Updated 5 months ago
- Pairwise Metrics for PyTorch☆18Updated 3 months ago
- Use Jax functions in Pytorch☆239Updated last year
- ☆32Updated 11 months ago
- A parallel ODE solver for PyTorch☆254Updated 6 months ago
- A tiny library for stochastic dataset caching in PyTorch.☆43Updated last year
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆306Updated last month