check https://github.com/arcee-ai/fastmlx
https://github.com/arcee-ai/fastmlx/pulls go through all PRs