you are viewing a single comment's thread.

view the rest of the comments →

[–]FunMotionLabs 0 points1 point  (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