piwheels

kim-jax

A package for developing knowledge informed inverse or forward mapping using ensemble learning

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install kim-jax

Dependencies

• hydroeval
• joblib
• tqdm
• scipy
• numpy
• seaborn
• pytest
• scikit-learn
• equinox
• ipykernel
• jupyter
• matplotlib
• pandas
• optax
• torchvision
• jax
• torch

Releases

Version	Released	Bullseye Python 3.9	Bookworm Python 3.11	Trixie Python 3.13	Files
1.3.0	2026-01-27
kim_jax-1.3.0-py3-none-any.whl (34 KB)
1.2.0	2025-07-31
kim_jax-1.2.0-py3-none-any.whl (42 KB)
1.0.0	2025-07-07
kim_jax-1.0.0-py3-none-any.whl (31 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue