It seems _quantize_mx() is not properly handling axes=None.
At
|
axes = [x + A.ndim if x < 0 else x for x in axes] |
return _quantize_mx(
File "c:\mx\microxcaling\mx_ops.py", line 194, in _quantize_mx
axes = [x + A.ndim if x < 0 else x for x in axes]
TypeError: 'NoneType' object is not iterable
Should not axes be initialized to shape of A if None?