v0.5.0

Highlights:

• Faster convolutions.
  • Up to 14x faster for some common sizes.
  • See benchmarks

Core

• mx.where properly handles inf
• Faster and more general convolutions
  • Input and kernel dilation
  • Asymmetric padding
  • Support for cross-correlation and convolution
• atleast_{1,2,3}d accept any number of arrays

NN

• nn.Upsample layer
  • Supports nearest neighbor and linear interpolation
  • Any number of dimensions

Optimizers

• Linear schedule and schedule joiner:
  • Use for e.g. linear warmup + cosine decay

Bugfixes

• arange throws on inf inputs
• Fix Cmake build with MLX
• Fix logsumexp inf edge case
• Fix grad of power w.r.t. to exponent edge case
• Fix compile with inf constants
• Bug temporary bug in convolution