Metadata-Version: 2.4
Name: wejax
Version: 1.0.2
Summary: Compute the Fisher Information Matrix (FIM) of your favorite JAX models
Project-URL: Repository, https://gitlab.com/lisamission/wejax
Project-URL: Documentation, https://lisamission.gitlab.com/wejax
Project-URL: Issues, https://gitlab.com/lisamission/wejax/-/issues
Author-email: Jean-Baptiste Bayle <j2b.bayle@gmail.com>, Maude Le Jeune <lejeune@apc.in2p3.fr>, Andrea Sartirana <sartirana@apc.in2p3.fr>
Maintainer-email: Jean-Baptiste Bayle <j2b.bayle@gmail.com>
License: BSD-3-Clause
License-File: LICENSE
Keywords: data analysis,fim,fisher,jax
Classifier: Topic :: Scientific/Engineering :: Astronomy
Classifier: Topic :: Scientific/Engineering :: Mathematics
Classifier: Topic :: Scientific/Engineering :: Physics
Requires-Python: <4,>=3.12
Requires-Dist: ipykernel>=6.29.5
Requires-Dist: ipython>=8.31.0
Requires-Dist: jax>=0.4.38
Requires-Dist: matplotlib>=3.10.0
Requires-Dist: numpy>=2.2.1
Requires-Dist: scipy>=1.15.0
Requires-Dist: sphinx-copybutton>=0.5.2
Requires-Dist: sympy>=1.13.3
Provides-Extra: tutorial
Requires-Dist: fastgb[jax]; extra == 'tutorial'
Requires-Dist: fomweb; extra == 'tutorial'
Requires-Dist: h5py>=3.12.1; extra == 'tutorial'
Requires-Dist: lisabeta>=1.0.2; extra == 'tutorial'
Requires-Dist: lisaconstants>=1.3.6; extra == 'tutorial'
Description-Content-Type: text/markdown

# 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:

```bash
poetry install
```

You can now run commands inside a dedicated virtual environment by running:

```bash
poetry run <your-command>
```
