Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

So in general it should work just installing the requirements.txt, however, we recommend installing TiRex inside the provided conda environment as we observe some people have problems with CUDA stuff otherwise. (TiRex uses custom CUDA kernels of the sLSTM)

I relaxed the major dependencies lock of the requirements.txt, and added a conda export of the environment on which i ran the test (freezed_test_environment.yaml)