ParentFull threadriyadparvez·Can you explain how JAX compares to pytorch? AFAIK pytorch also closely resembles numpy API.View on HN