Franck Verrot 7df3f792a2 Ensure Conv2D and Conv3D's kernel sizes aren't trimmed (#1852) ... Before the change, this snippet: ``` print(nn.Conv1d(1, 32, 3, padding=1)) print(nn.Conv2d(1, 32, (3, 3), padding=1)) print(nn.Conv3d(1, 32, (3, 3, 3), padding=1)) ``` would output: ``` Conv1d(1, 32, kernel_size=3, stride=1, padding=1, dilation=1, groups=1, bias=True) Conv2d(1, 32, kernel_size=(3,), stride=(1, 1), padding=(1, 1), dilation=1, groups=1, bias=True) Conv3d(1, 32, kernel_size=(3, 3), stride=(1, 1, 1), padding=(1, 1, 1), dilation=1, bias=True) ``` After the change, the output will be: ``` Conv1d(1, 32, kernel_size=3, stride=1, padding=1, dilation=1, groups=1, bias=True) Conv2d(1, 32, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), dilation=1, groups=1, bias=True) Conv3d(1, 32, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1), dilation=1, bias=True) ```		2025-02-10 06:27:01 -08:00
..
nn	Ensure Conv2D and Conv3D's kernel sizes aren't trimmed (#1852)	2025-02-10 06:27:01 -08:00
optimizers	Support bias correction in Adam and AdamW optimizers (#1640)	2024-12-06 12:13:34 -08:00
__main__.py	MLX in C++ example (#1736)	2025-01-02 19:09:04 -08: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
_stub_patterns.txt	fix submodule stubs (#1492)	2024-10-15 16:23:37 -07:00
distributed_run.py	allow command (#1836)	2025-02-06 10:32:24 -08: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