Comment by thatguysaguy

The claim was that Python is only used for prototyping at Google, not that people are writing the AI frameworks themselves in pure Python.

JAX is obviously implemented in C++, but the scientists running the training runs which cost millions of dollars are writing lots of python.