4 ms·
If you like PyTorch then you might like Equinox, by the way. (https://github.com/patrick-kidger/equinox https://github.com/patrick-kidger/equinox ; 1.4k GitHub
by patrickkidger 3y ago
If you like PyTorch then you might like Equinox, by the way. (https://github.com/patrick-kidger/equinox https://github.com/patrick-kidger/equinox ; 1.4k GitHub stars now!)
Basically designed to offer PyTorch-like syntax for working with JAX. The latter is excellent for the reasons the sibling replies have stated, but PyTorch absolutely got the usability story correct.