Installation
pip install parametrix
Example
The following example shows how to use Param as a base class for a parameter class that enforces positivity:
import jax.numpy as jnp
from parametrix import Param
class PositiveOnlyParam(Param):
def __init__(self, value):
super().__init__(jnp.log(value))
@property
def value(self):
return jnp.exp(self.raw_value)
The backing values of Params are always stored as jax.Arrays, meaning that they will automatically be picked up as learnable parameters by libraries like Equinox.
Param objects also behave like numeric types, so that they are able to be used within models and any other functions without having to make any changes to the code.
Documentation
API documentation is available at Read the Docs.
1 maintainer
Metadata
1 maintainer
Release files for parametrix 0.1.8
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| parametrix-0.1.8.tar.gz | 3.0 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| parametrix-0.1.8-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 6.5 kB
Release files / parametrix-0.1.8.tar.gz
| Download URL | parametrix-0.1.8.tar.gz |
|---|---|
| Size | 3.0 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
77795de04f4c668b1c2c9bdc5984e27287ab0d6c5a120de947126d526bd3897b
|
|
BLAKE2b-256 checksum How to use checksums |
6eec70166330d082dbef5049b50556124f5f30eb064604ec37fedacede92bfda
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
uv/0.12.9 {"installer":{"name":"uv","version":"0.12.9","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
|
Release files / parametrix-0.1.8-py3-none-any.whl
| Download URL | parametrix-0.1.8-py3-none-any.whl |
|---|---|
| Size | 3.5 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
6157a31b4fa6d6f5e8eabe38dd1f55b282787fafca4fd0d9de11fc75cffedf60
|
|
BLAKE2b-256 checksum How to use checksums |
35a553f0bc374e31b580dbeee0173dbcf7f5a39e1dd09f27f9be6a16be78dd3f
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
uv/0.12.9 {"installer":{"name":"uv","version":"0.12.9","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
|
