Daniel Strobusch 479051ce1c add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427) ...

* add numeric type hierarchy and issubdtype as well as a set_dtype method to nn.Module with predicate

numeric type hierarchy and issubtype is compatible to the [numpy hierarchy](220f0ab2c5/numpy/_core/numerictypes.py (L42)).

Closes #285.

* nits in docs

* unify type category checking

* nits in docs

* nits in docs

* more docs nits

* fix callable type

---------

Co-authored-by: Awni Hannun <awni@apple.com>

2024-03-25 12:32:59 -07:00

..

_static

Change the logo and add a dark option (#716)

2024-02-20 10:57:02 -08:00

_templates

Compile with capture (#629)

2024-02-07 17:29:22 -08:00

cpp

jagrit's commit files

2023-11-29 10:52:08 -08:00

dev

Fixed wording in extensions.rst (#678)

2024-02-13 08:39:02 -08:00

examples

Update mlx-example link for llms llama in llama-inference.rst (#405)

2024-01-08 16:29:53 -08:00

python

add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)

2024-03-25 12:32:59 -07:00

usage

Update function_transforms.rst (#796)

2024-03-06 12:03:37 -08:00

conf.py

post nanobind docs fixes and some updates (#889)

2024-03-24 15:03:27 -07:00

index.rst

add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427)

2024-03-25 12:32:59 -07:00

install.rst

Switch to nanobind (#839)

2024-03-18 20:12:25 -07:00