mlx/python/mlx
Max-Heinrich Laves efeb9c0f02
Transposed Convolution (#1245)
* initial implementation for conv_transpose

ran pre-commit

implemented conv_transpose

updated conv_general docstring

updated conv_general docstring

updated code comments

removed commented run_conv_checks

updated acknowledgments

added missing entry to ops.rst

added op to nn.layers

resolved merge conflicts

* removed ConvolutionTranspose primitive as suggested by reviewer

removed ConvolutionTranspose primitive as suggested by reviewer

* remove transpose flag, add another test

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2024-09-06 19:52:38 -07:00
..
nn Transposed Convolution (#1245) 2024-09-06 19:52:38 -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 copyright + ack 2023-11-30 11:12:53 -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