Lucas Newman 4a64d4bff1 Add support for grouped 1D convolutions to the nn API (#1444) ... * Fix the weight shape for grouped convolutions from the nn API. * Add tests. * Pre-commit formatting. * Add input validation. * Use integer division instead of casting. * docs * nit --------- Co-authored-by: Awni Hannun <awni@apple.com>		2024-09-28 06:41:07 -07:00
..
nn	Add support for grouped 1D convolutions to the nn API (#1444)	2024-09-28 06:41:07 -07:00
optimizers	Some fixes to typing (#1371)	2024-08-28 11:16:19 -07:00
_os_warning.py	Version bump and os error (#807)	2024-03-07 13:57:58 -08:00
_reprlib_fix.py	override class function (#1418)	2024-09-16 13:21:04 -07:00
extension.py	Adding missing type hints (#1243)	2024-07-23 07:29:38 -07: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	In continuation to PR1243 to solve issue #1240 (#1365)	2024-08-28 11:40:41 -07:00