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