Add support for _expected_input_spec in MLX Lambda to match JAX implementation#17
Open
lancelotblanchard wants to merge 1 commit into
Open
Add support for _expected_input_spec in MLX Lambda to match JAX implementation#17lancelotblanchard wants to merge 1 commit into
lancelotblanchard wants to merge 1 commit into