Escalante uses JAX on TPUs for AI-driven protein design
As a Python library for accelerator-oriented array computation and program transformation, JAX is widely recognized ...
google cloudblog