3 ms·

I have a preference for Flax, you basically get Pytorch but streamlined thanks to the Jax foundation.

• SleekEagle 5y ago
  Awesome, thanks for the suggestion. I'll check out both for sure!