torch-engression

GPU-accelerated distributional regression via energy scores.

See the examples page for the benchmark notebook and generated figures.