TorchOptimizer function

Torch Optimizer