Add improvements to Sampler class

[kmkolasinski]

          pip install uv
          uv venv
          source .venv/bin/activate
          sudo apt-get install -y libturbojpeg-dev
          uv pip install -e ".[dev,datasets]"