Flax vs. TensorFlow

[MWITIDERRICK]

Flax vs. TensorFlow

[8bitmp3]

@mwitiderrick Some of the recently added new Flax docs, if you’re interested:

• Scale up Flax Modules on multiple devices
• Dropout
• Batch normalization
• Save and load checkpoints