Swift для дата-сайентиста: быстрое погружение за 2 часа

Если очень не хочется писать TensorFlow код на питоне, можно пользоваться обёрткой для Julia, которая тоже позволяет не писать tf.while_loop, заменяя эту магию на макрос @tf.

Код: https://github.com/malmaud/TensorFlow.jl
И поддержка CUDA есть.