Nathan

I'd say we should reopen this. That's not a unit test, and adding unit tests would add value to this project.

Does lowering your learning rate help?

I just realized that pull request #45 addresses this same issue. I believe our approaches are a bit different, so it may be worth looking in to both to see...

I can try to take a look at it too. I'll let you know if I get anywhere.

See pull request #95

I am seeing the same error in 2.0.1: ```bash ❯ python -c "import torch;print(torch.__version__);t=torch.ones(2,6,device='mps')[1].reshape(2, 3); t = t+1; print(t)" 2.0.1 /AppleInternal/Library/BuildRoots/2acced82-df86-11ed-9b95-428477786501/Library/Caches/com.apple.xbs/Sources/MetalPerformanceShaders/MPSCore/Types/MPSNDArray.mm:84: failed assertion `[MPSNDArrayDescriptor sliceDimension:withSubrange:] error: subRange.start (6) is not...