Chunyang Wen

Fix bug in assign_op.h, Use not existed variables

1

Code in assign_op.h uses wrong variables. assign_op.h is included in parallel_kv_match.h. But it seems that parallel_kv_match.h is not used.

Remove unused import

6

## Proposed changes Remove unused import. ## Checklist Put an `x` in the boxes that apply. - [x] I have read the [CONTRIBUTING](https://github.com/ml-explore/mlx/blob/main/CONTRIBUTING.md) document - [x] I have run `pre-commit...

cannot install on Apple Silicon

2

I am following the doc here: https://github.com/apple/axlearn/blob/a7a738e816363ff4950b0da391206c3216e87d6d/docs/01-start.md Failed on the final step: ```bash pip install "axlearn[core, apple-silicon]" ``` ```bash ERROR: Ignored the following versions that require a different python version:...