jax implementation of robust heteroskedastic matrix factorisation. Robusta like the coffee bean, get it?
Easiest is from PyPI either with pip
pip install robusta-hmfor uv (recommended)
uv add robusta-hmfOr, you can clone and build from source
git clone git@github.com:TomHilder/robusta-hmf.git
cd robusta-hmf
pip install -e .TODO
TODO
TODO
- More interpretable loss, maybe normalised in some sensible way (maybe proper NLL)
- Type checking with
mypy