Installation

Basic Install

pip install torch_measure

Optional Dependencies

# Bayesian estimation (Pyro SVI)
pip install torch_measure[bayesian]

# Visualization (matplotlib, seaborn)
pip install torch_measure[viz]

# Data loaders (HuggingFace Hub)
pip install torch_measure[data]

# Everything
pip install torch_measure[all]

Development Install

git clone https://github.com/aims-foundations/torch_measure.git
cd torch_measure
pip install -e ".[dev,test]"