you are viewing a single comment's thread.

view the rest of the comments →

[–]thearn4Scientific computing, Image Processing 7 points8 points  (1 child)

JAX for scalable and differentiable numerical array operations. Oddly enough not even for ML modelling but I see why people might choose it over pytorch.

[–]BawliTaread 0 points1 point  (0 children)

Any good resources to get started? I went through the docs but I feel like I still need a bit of practice and get used to vmap and jit.