v0.4.0

Highlights:

• Partial shapeless compilation
  • Default shapeless compilation for all activations
  • Can be more than 5x faster than uncompiled versions
• CPU kernel fusion
  • Some functions can be up to 10x faster

Core

• CPU compilation
• Shapeless compilation for some cases
  • mx.compile(function, shapeless=True)
• Up to 10x faster scatter: benchmarks
• mx.atleast_1d, mx.atleast_2d, mx.atleast_3d

Bugfixes

• Bug with tolist with bfloat16 and float16
• Bug with argmax on M3