3 ms·

Google has moved to JAX. I know many people who prefer it over pytorch.

• leoh 1y ago
  It’s okay. Complaints are documentation, limited community support (all kinds of architecture is much more diy for it vs PyTorch). Unrelated gripe: they architected it really poorly from a pure sw pov imo. Specifically it’s all about Python bindings for C++ so the py/c++ layer is tightly coupled both in code and in the build system. They have a huge opportunity to fix this so, for example, rust bindings could be (reasonably trivially) generated, let alone for other languages.