Skip to main content

Differentiable minimization in jax using Newton's method

v0.0.1

This project essentially repackages code from the implicit layers tutorial to provide a minimize_newton function.

Given a function fn(params, z), it finds the z_star which minimizes fn for given params. Further, the gradient of the solution with respect to params can be computed; this is done using a custom vjp rule, as shown in the tutorial.

Installation

mewtax can be installed via pip:

pip install mewtax

Metadata

Release files for mewtax 0.0.1

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for mewtax 0.0.1
File Size Uploaded Provenance Actions
mewtax-0.0.1.tar.gz 6.7 kB Without provenance Details for mewtax-0.0.1.tar.gz

Built distribution (wheel)

Table of built distributions (wheels) for mewtax 0.0.1
File Interpreter ABI Platform Actions
mewtax-0.0.1-py3-none-any.whl Python 3 none any Details for mewtax-0.0.1-py3-none-any.whl

Total release size: 12.2 kB

Release files / mewtax-0.0.1.tar.gz

Download URL mewtax-0.0.1.tar.gz
Size 6.7 kB
Tags Source
SHA-256 checksum
How to use checksums
601d9acb23b979403ce1a954cdff20fa485dc99434f8e0d2560eac5de8cff396
BLAKE2b-256 checksum
How to use checksums
08497d5883333d201caed32d898ef58821f8a2c51c9bbb492ca2b37be855d926
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/5.1.0 CPython/3.12.3

Release files / mewtax-0.0.1-py3-none-any.whl

Download URL mewtax-0.0.1-py3-none-any.whl
Size 5.5 kB
Tags Python 3
SHA-256 checksum
How to use checksums
2c42d16d257137d06c4f5a02dd3b144c18cd53f2cc5592fa5c2b2e069a3004fe
BLAKE2b-256 checksum
How to use checksums
8ae4d745634168cd43df9bdea7269c91d7edae67cd0752dda8d2c4281981a144
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/5.1.0 CPython/3.12.3

Security for mewtax 0.0.1

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Release history Release notifications | RSS feed

This release

0.0.1 This release

2 release files No provenance

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page