torch stub missing many methods

### Describe the Bug

https://github.com/open-spaced-repetition/srs-benchmark/actions/runs/25816239178/job/75845375535?pr=315

some missing methods:

Error: No attribute `square` in module `torch`
Error: No attribute `square` in module `torch`
Error: No attribute `equal` in module `torch`
Error: Object of class `Tensor` has no attribute `nonzero`
Error: Unexpected keyword argument `device` in function `torch.ones_like`
Error: No attribute `square` in module `torch`
Error: No attribute `square` in module `torch`
Error: No attribute `equal` in module `torch`
Error: Object of class `Tensor` has no attribute `nonzero`
Error: Unexpected keyword argument `device` in function `torch.ones_like`

Error: No attribute `RNN` in module `torch.nn`
Error: No attribute `load` in module `torch`
Error: No attribute `Transformer` in module `torch.nn`
Error: No attribute `load` in module `torch`
Error: No attribute `load` in module `torch`

Error: No attribute `load` in module `torch`
Error: No attribute `one_hot` in module `torch.nn.functional`
Error: No attribute `load` in module `torch`
Error: No attribute `RNN` in module `torch.nn`

### Sandbox Link

_No response_

### (Only applicable for extension issues) IDE Information

_No response_