integrate pytorch

[liellnima]

integrate parts of Daria's pytorch code, handle dependencies accordingly

[liellnima]

very big task, needs to be broken down and we need to create subissues for that one