mlx/docs/src/python
Nikhil Mehta 0b7d71fd2f
Add softmin, hardshrink, hardtanh (#1180)
---------

Co-authored-by: Nikhil Mehta <nikmehta@tesla.com>
2024-06-04 15:48:18 -07:00
..
nn Add softmin, hardshrink, hardtanh (#1180) 2024-06-04 15:48:18 -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 view op (#1179) 2024-06-04 08:05: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