Fix for type mismatch bug and numerical stability improvements#449

shantanusharma wants to merge 3 commits intoTorchSim:mainTorchSim/torch-sim:mainfrom

shantanusharma:mainshantanusharma/torch-sim:mainCopy head branch name to clipboard