Piotr Rybiec 6a665ea6ed Dilation for convolutional layers (#766) ... * add dilation parameter to Conv1d layer * space here too * add conv1d dilation test * add dilation parameter for Conv2d layer * conv2d dilation test		2024-03-04 06:43:00 -08:00
..
nn	Dilation for convolutional layers (#766)	2024-03-04 06:43:00 -08:00
optimizers	Add linear warmup and schedule joining for use with existing schedules (#721)	2024-02-26 07:28:48 -08:00
_reprlib_fix.py	copyright + ack	2023-11-30 11:12:53 -08:00
extension.py	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
py.typed	Add py.typed to support PEP-561 (type-hinting) for mlx (#588)	2024-01-31 12:05:42 -08:00
utils.py	Adds device context manager (#679)	2024-02-14 14:14:58 -08:00