mlx/python/mlx/_stub_patterns.txt
2024-10-15 16:23:37 -07:00

21 lines
689 B
Plaintext

mlx.core.distributed.__prefix__:
from mlx.core import array, Dtype, Device, Stream
from mlx.core.distributed import Group
from typing import Sequence, Optional, Union
mlx.core.fast.__prefix__:
from mlx.core import array, Dtype, Device, Stream
from typing import Sequence, Optional, Union
mlx.core.linalg.__prefix__:
from mlx.core import array, Dtype, Device, Stream
from typing import Sequence, Optional, Tuple, Union
mlx.core.metal.__prefix__:
from mlx.core import array, Dtype, Device, Stream
from typing import Sequence, Optional, Union
mlx.core.random.__prefix__:
from mlx.core import array, Dtype, Device, Stream
from typing import Sequence, Optional, Union