Compute the Fisher Information Matrix (FIM) of your favorite JAX models
Project description
WeJAX
Compute the Fisher Information Matrix (FIM) of your favorite models using their Jacobians; take advantage of automatic differentiation if your model is implemented in JAX to switly compute the Jacobian, or evaluate it numerically with finite differences.
WeJAX also provide convenience fonctions to invert the FIM and compute error bars and correlations on the estimated parameters, as well as nice-looking plots.
Contributing
This project uses Poetry for dependency management. To install the dependencies and the project itself, run the following command:
poetry install
You can now run commands inside a dedicated virtual environment by running:
poetry run <your-command>
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file wejax-1.0.2.tar.gz.
File metadata
- Download URL: wejax-1.0.2.tar.gz
- Upload date:
- Size: 941.5 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.11.1 {"installer":{"name":"uv","version":"0.11.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Debian GNU/Linux","version":"13","id":"trixie","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
6e8ff97525a850faa562d482dc611c6c136a70632c433ddff96f88b8b4ce3ac5
|
|
| MD5 |
f953ec7f2af54c04fdd36e1e4bc8f523
|
|
| BLAKE2b-256 |
b0680e482e55e333b70db99b250ef327fa9199d48c707f9bca73b1c31fdf76b1
|
File details
Details for the file wejax-1.0.2-py3-none-any.whl.
File metadata
- Download URL: wejax-1.0.2-py3-none-any.whl
- Upload date:
- Size: 13.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.11.1 {"installer":{"name":"uv","version":"0.11.1","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Debian GNU/Linux","version":"13","id":"trixie","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":true}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
53301128f783029d8789eb1b130c7302fcde7771573f1ef64710afbd9407a2f4
|
|
| MD5 |
b722bd6a61d72bdac7759271ce108430
|
|
| BLAKE2b-256 |
d8a321c6c4ab2c82c8da0e3dee69443765154fb66dfed9e89ac11bee652ef7f0
|
