All pull requests

Pull requests

• Add support for _expected_input_spec in MLX Lambda to match JAX implementation

  #17·lancelotblanchard opened Jun 22, 2026·

  ·
  1

  

• Fix MLX implementation of LocalDotProductSelfAttention that previously overwrote block_size

  #16·lancelotblanchard opened Jun 10, 2026·

  ·
  2