Seems the package is not actively maintained, as before the release of 1.2.0 in 2025, it was a release in 2023, nearly 2 years for a single update but breaks soon. Perhaps we could consider Torch instead, which at least has mps support built-in and work well.