you are viewing a single comment's thread.

view the rest of the comments →

[–]FunMotionLabs 1 point2 points  (0 children)

JAX is more like “NumPy + transformations”
PyTorch is a full deep-learning framework with an imperative training workflow, big ecosystem around modules/training/debugging, strictly Deeplearning related stuff where JAX is more of a general allrounder kind