adds SAE gpu training