[BUG] Missing types

[marcospgp]

There are many missing types in the library, with many functions having Unknown parameter or return types:

I think it would be a great improvement to have full typing information across the MLX API.

[awni]

I will make this a feature and leave it open. I think it's a good idea to have useful type hints for the stable Python API.

[AbhinavMir]

I ran a quick Python script to see which files are missing type annotations, does this look right?

Report: https://pastebin.com/raw/PepZW2jM

[awni]

I would only consider files in the subdirectory python/mlx for now. We don't need to add strict type hints to benchmarks and tests.

[AbhinavMir]

That makes sense - I've modified this here. I was just crawling .py files, changed that around.

https://github.com/AbhinavMir/mlx/blob/typeAnno/type_annotation_report.md

[ManishAradwad]

I have added some type hints for few functions given in above file by @AbhinavMir

Please let me know if they're right

[AbhinavMir]

One of my friends @catplotlib wants to try Open Source, tagging her here so she can pick up a few type hints!

[Jonathan-Dobson]

I second the motion to add typings.
The pip dist of mlx.core does not appear to offer type hints.

[jbcoe]

I added #1345 for missing type hints in pip.

[Saanidhyavats]

The issue is partially solved through #1243. Since the PR is inactive for more than a month, I will be continuing the work and create a new pr for updated changes.