mlx/docs/src/python
Abe Leininger 79ef49b2c2
add mx.trace (#1143) (#1147)
* working c++ trace implementation

* updated throw + added overloads

* added python binding for trace function

* pre-commit reformatting

* add trace to docs

* resolve comments

* remove to_stream call
2024-05-22 15:50:27 -07:00
..
nn Conv3d (#993) 2024-05-11 06:15:02 -07:00
optimizers Add linear warmup and schedule joining for use with existing schedules (#721) 2024-02-26 07:28:48 -08:00
array.rst add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427) 2024-03-25 12:32:59 -07:00
data_types.rst add numeric type hierarchy and issubdtype as well as a set_dtype meth… (#427) 2024-03-25 12:32:59 -07:00
devices_and_streams.rst Add synchronize function (#1006) 2024-04-22 08:25:46 -07:00
fast.rst post nanobind docs fixes and some updates (#889) 2024-03-24 15:03:27 -07:00
fft.rst awni's commit files 2023-11-29 10:30:41 -08:00
linalg.rst Some fixes in docs (#1141) 2024-05-20 11:51:47 -07:00
metal.rst Reset peak memory (#1074) 2024-05-03 17:12:51 -07:00
nn.rst Quantize embedding (#994) 2024-04-15 16:42:10 -07:00
ops.rst add mx.trace (#1143) (#1147) 2024-05-22 15:50:27 -07:00
optimizers.rst feat: implement clip_grad_norm (#1043) 2024-05-03 09:07:02 -07:00
random.rst Implementation of mlx.random.multivariate_normal (#502) (#877) 2024-04-09 13:50:12 -07:00
transforms.rst Compile docs (#653) 2024-02-08 11:21:50 -08:00
tree_utils.rst feat: implement clip_grad_norm (#1043) 2024-05-03 09:07:02 -07:00