v0.2.0

Highlights:

• mx.compile makes stuff go fast
  • Some functions are up to 10x faster (benchmarks)
  • Training models anywhere from 10% to twice as fast (benchmarks)
  • Simple syntax for compiling full training steps

Core

• mx.compile function transformation
• Find devices properly for iOS
• Up to 10x faster GPU gather
• __abs__ overload for abs on arrays
• loc and scale in parameter for mx.random.normal

NN

• Margin ranking loss
• BCE loss with weights

Bugfixes

• Fix for broken eval during function transformations
• Fix mx.var to give inf with doff >= nelem
• Fix loading empty modules in nn.Sequential