Pytorch is indeed used for many models and is a useful open framework. Most of our open models use either @TensorFlow or JAX, which are Google-initiated open frameworks and are also heavily used.
Google’s Open Frameworks: TensorFlow and JAX for Model Development
By
–