Skip to main content

Basic PyTorch implementation of the Levenberg-Marquardt algorithm

Project description

levmarq-torch

A basic PyTorch implementation of the Levenberg-Marquardt algorithm. This solves minimization problems of the form

$$\mathbf{x}^* = \mathrm{argmin}_{\mathbf{x}} |\mathbf{y} - \mathbf{\hat{y}}(\mathbf{x})|^2 , .$$

The implementation is batched over the parameters $\mathbf{x}$ and datapoints $\mathbf{y}$.

Based on implementation 1 from Gavin 2022 and some help from Connor Stone.

Installation

Running

git clone git@github.com:adam-coogan/levmarq-torch.git
cd levmarq-torch
pip install .

will install the levmarq_torch package.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

levmarq_torch-0.0.1.tar.gz (5.0 kB view hashes)

Uploaded Source

Built Distribution

levmarq_torch-0.0.1-py3-none-any.whl (5.7 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page