Publish package on PyPI

Hello! I just randomly stumbled onto your repo and I was curious as to whether you have any plans to publish it to PyPI.

My use case is a bit basic - I'm currently playing around with writing a small nanochat-esque clone repo for training hybrid models and I was looking for reference mlx implementations of fla that I can directly reuse rather than having to vibe code one. I found yours and I was hoping to directly `pip install` your project rather than doing a manual setup.

I am happy to help contribute in any way to getting your package published including writing documentation or dogfooding implementations!

No files, tests, or packaging entry points are named. Start by inspecting the repository's packaging metadata and installation instructions, then determine what is needed for a PyPI release; consider the work done when the project can be installed from PyPI with the documented setup.

Written by the indexing model from the issue text.