MCPcopy Create free account

hub / github.com/ml-explore/mlx / functions

Functions4,946 in github.com/ml-explore/mlx

⨍ Functions 4,946◇ Types & classes 770↳ Endpoints 5
Method__call__
(self, x)
python/tests/test_export_import.py:382
Method__call__
(self, x)
python/tests/test_export_import.py:406
Method__call__
(self, x)
python/tests/test_export_import.py:432
Method__call__
(self, x)
python/tests/mlx_distributed_tests.py:275
Method__call__
(self, x)
examples/export/eval_mlp.py:21
Method__call__
(self, x)
examples/export/train_mlp.py:22
Method__call__
(self, input)
benchmarks/python/conv3d_train_bench_cpu.py:34
Method__call__
(self, input)
benchmarks/python/conv2d_train_bench_cpu.py:34
Method__getattr__
(self, key: str)
python/mlx/nn/layers/base.py:99
Method__init__
(self, name: str, sourcedir: str = "")
setup.py:65
Method__init__
(self, name: str, sourcedir: str = "")
python/mlx/extension.py:19
Method__init__
(self, *modules)
python/mlx/nn/layers/containers.py:17
Method__init__
( self, in_channels: int, out_channels: int, kernel_size: Union[int, tuple],
python/mlx/nn/layers/convolution_transpose.py:109
Method__init__
( self, in_channels: int, out_channels: int, kernel_size: Union[int, tuple],
python/mlx/nn/layers/convolution_transpose.py:189
Method__init__
( self, num_embeddings: int, dims: int, group_size: int = None, bits:
python/mlx/nn/layers/quantized.py:117
Method__init__
( self, input_dims: int, output_dims: int, bias: bool = True, group_si
python/mlx/nn/layers/quantized.py:223
Method__init__
(self, *args: Any, **kwargs: Any)
python/mlx/nn/layers/linear.py:19
Method__init__
( self, input1_dims: int, input2_dims: int, output_dims: int, bias: bool = True )
python/mlx/nn/layers/linear.py:135
Method__init__
( self, input_size: int, hidden_size: int, bias: bool = True, )
python/mlx/nn/layers/recurrent.py:123
Method__init__
( self, input_size: int, hidden_size: int, bias: bool = True, )
python/mlx/nn/layers/recurrent.py:234
Method__init__
( self, scale_factor: Union[float, Tuple], mode: Literal["nearest", "linear", "cubic"]
python/mlx/nn/layers/upsample.py:228
Method__init__
Should be called by the subclasses of ``Module``.
python/mlx/nn/layers/base.py:61
Method__init__
(self, p: float = 0.5)
python/mlx/nn/layers/dropout.py:61
Method__init__
(self, p: float = 0.5)
python/mlx/nn/layers/dropout.py:110
Method__init__
( self, input_dims: int, output_dims: int, bias: bool = True, group: O
python/mlx/nn/layers/distributed.py:293
Method__init__
( self, input_dims: int, output_dims: int, bias: bool = True, group_si
python/mlx/nn/layers/distributed.py:381
Method__init__
( self, input_dims: int, output_dims: int, bias: bool = True, group_si
python/mlx/nn/layers/distributed.py:516
Method__init__
( self, dims: int, min_freq: float = 0.0001, max_freq: float = 1, scal
python/mlx/nn/layers/positional_encoding.py:77
Method__init__
( self, pooling_function, padding_value, kernel_size: Union[int, Tuple[int]],
python/mlx/nn/layers/pooling.py:114
Method__init__
( self, pooling_function, padding_value, kernel_size: Union[int, Tuple[int, in
python/mlx/nn/layers/pooling.py:138
Method__init__
( self, pooling_function, padding_value, kernel_size: Union[int, Tuple[int, in
python/mlx/nn/layers/pooling.py:162
Method__init__
( self, kernel_size: Union[int, Tuple[int]], stride: Optional[Union[int, Tuple[int]]]
python/mlx/nn/layers/pooling.py:207
Method__init__
( self, kernel_size: Union[int, Tuple[int]], stride: Optional[Union[int, Tuple[int]]]
python/mlx/nn/layers/pooling.py:238
Method__init__
( self, kernel_size: Union[int, Tuple[int, int]], stride: Optional[Union[int, Tuple[in
python/mlx/nn/layers/pooling.py:276
Method__init__
( self, kernel_size: Union[int, Tuple[int, int]], stride: Optional[Union[int, Tuple[in
python/mlx/nn/layers/pooling.py:314
Method__init__
( self, kernel_size: Union[int, Tuple[int, int, int]], stride: Optional[Union[int, Tup
python/mlx/nn/layers/pooling.py:353
Method__init__
( self, kernel_size: Union[int, Tuple[int, int, int]], stride: Optional[Union[int, Tup
python/mlx/nn/layers/pooling.py:392
Method__init__
(self, negative_slope=1e-2)
python/mlx/nn/layers/activations.py:414
Method__init__
(self, alpha=1.0)
python/mlx/nn/layers/activations.py:432
Method__init__
(self, lambd=0.5)
python/mlx/nn/layers/activations.py:473
Method__init__
(self, alpha=1.0)
python/mlx/nn/layers/activations.py:492
Method__init__
(self, num_parameters=1, init=0.25)
python/mlx/nn/layers/activations.py:536
Method__init__
(self, approx="none")
python/mlx/nn/layers/activations.py:572
Method__init__
(self, threshold: float = 0.0)
python/mlx/nn/layers/activations.py:621
Method__init__
( self, in_channels: int, out_channels: int, kernel_size: Union[int, tuple],
python/mlx/nn/layers/convolution.py:110
Method__init__
( self, in_channels: int, out_channels: int, kernel_size: Union[int, tuple],
python/mlx/nn/layers/convolution.py:189
Method__init__
( self, dims: int, num_heads: int, query_input_dims: Optional[int] = None,
python/mlx/nn/layers/transformer.py:48
Method__init__
( self, dims: int, num_heads: int, mlp_dims: Optional[int] = None, dro
python/mlx/nn/layers/transformer.py:104
Method__init__
( self, num_layers: int, dims: int, num_heads: int, mlp_dims: Optional
python/mlx/nn/layers/transformer.py:154
Method__init__
( self, dims: int, num_heads: int, mlp_dims: Optional[int] = None, dro
python/mlx/nn/layers/transformer.py:183
Method__init__
( self, num_layers: int, dims: int, num_heads: int, mlp_dims: Optional
python/mlx/nn/layers/transformer.py:245
Method__init__
( self, dims: int, eps: float = 1e-5, affine: bool = False, )
python/mlx/nn/layers/normalization.py:42
Method__init__
( self, dims: int, eps: float = 1e-5, affine: bool = True, bias: bool = True )
python/mlx/nn/layers/normalization.py:93
Method__init__
( self, num_groups: int, dims: int, eps: float = 1e-5, affine: bool =
python/mlx/nn/layers/normalization.py:176
Method__init__
( self, num_features: int, eps: float = 1e-5, momentum: float = 0.1, a
python/mlx/nn/layers/normalization.py:281
Method__init__
(self, num_embeddings: int, dims: int)
python/mlx/nn/layers/embedding.py:23
Method__init__
(self, schedulers=None)
python/mlx/optimizers/optimizers.py:15
Method__init__
(self, optimizers, filters: list = [])
python/mlx/optimizers/optimizers.py:172
Method__init__
( self, learning_rate: Union[float, Callable[[mx.array], mx.array]], alpha: float = 0.
python/mlx/optimizers/optimizers.py:315
Method__init__
( self, learning_rate: Union[float, Callable[[mx.array], mx.array]], eps: float = 1e-8
python/mlx/optimizers/optimizers.py:372
Method__init__
( self, learning_rate: Union[float, Callable[[mx.array], mx.array]], rho: float = 0.9,
python/mlx/optimizers/optimizers.py:425
Method__init__
( self, learning_rate: Union[float, Callable[[mx.array], mx.array]], betas: List[float
python/mlx/optimizers/optimizers.py:493
Method__init__
( self, learning_rate: Union[float, Callable[[mx.array], mx.array]], betas: List[float
python/mlx/optimizers/optimizers.py:564
Method__init__
( self, learning_rate: Union[float, Callable[[mx.array], mx.array]], betas: List[float
python/mlx/optimizers/optimizers.py:615
Method__init__
( self, learning_rate: Union[float, Callable[[mx.array], mx.array]], betas: List[float
python/mlx/optimizers/optimizers.py:677
Method__init__
( self, learning_rate: Union[float, Callable[[mx.array], mx.array], None] = None, eps:
python/mlx/optimizers/optimizers.py:742
Method__init__
( self, learning_rate: Union[float, Callable[[mx.array], mx.array]], momentum: float =
python/mlx/optimizers/optimizers.py:876
Method__init__
(self, hosts, tb_hosts, uuid_reverse_index)
python/mlx/_distributed_utils/config.py:124
Method__init__
(self, rank, host, python, cwd, files, env, command)
python/mlx/_distributed_utils/launch.py:48
Method__init__
(self)
python/tests/test_optimizers.py:536
Method__init__
(self)
python/tests/test_nn.py:78
Method__init__
(self)
python/tests/test_nn.py:307
Method__init__
(self)
python/tests/test_nn.py:325
Method__init__
(self, *args, **kwargs)
python/tests/mlx_tests.py:21
Method__init__
(self)
python/tests/test_export_import.py:320
Method__init__
(self)
python/tests/test_export_import.py:378
Method__init__
(self)
python/tests/test_export_import.py:402
Method__init__
(self)
python/tests/test_export_import.py:426
Method__init__
(self, *args, **kwargs)
python/tests/mlx_distributed_tests.py:270
Method__init__
( self, num_layers: int, input_dim: int, hidden_dim: int, output_dim: int )
examples/export/eval_mlp.py:11
Method__init__
( self, num_layers: int, input_dim: int, hidden_dim: int, output_dim: int )
examples/export/train_mlp.py:12
Method__init__
(self, in_channels, hidden_channels=16)
benchmarks/python/conv3d_train_bench_cpu.py:79
Method__init__
(self, in_channels, hidden_channels=32)
benchmarks/python/conv2d_train_bench_cpu.py:79
Method__repr__
(self)
python/mlx/nn/layers/base.py:87
Method__setattr__
(self, key: str, val: Any)
python/mlx/nn/layers/base.py:105
Function_bs_qmm_dispatch_typed
mlx/backend/cpu/quantized.cpp:701
Function_check
(hostname, i)
python/mlx/_distributed_utils/config.py:373
Function_cubic_indices
(N, scale, align_corners, dim, ndims)
python/mlx/nn/layers/upsample.py:54
Method_extra_repr
(self)
python/mlx/nn/layers/convolution_transpose.py:61
Method_extra_repr
(self)
python/mlx/nn/layers/convolution_transpose.py:140
Method_extra_repr
(self)
python/mlx/nn/layers/convolution_transpose.py:222
Method_extra_repr
(self)
python/mlx/nn/layers/quantized.py:173
Method_extra_repr
(self)
python/mlx/nn/layers/quantized.py:257
Method_extra_repr
(self)
python/mlx/nn/layers/quantized.py:360
Method_extra_repr
(self)
python/mlx/nn/layers/linear.py:63
Method_extra_repr
(self)
python/mlx/nn/layers/linear.py:152
Method_extra_repr
(self)
python/mlx/nn/layers/recurrent.py:68
Method_extra_repr
(self)
python/mlx/nn/layers/recurrent.py:150
Method_extra_repr
(self)
python/mlx/nn/layers/recurrent.py:256
Method_extra_repr
(self)
python/mlx/nn/layers/upsample.py:244
← previousnext →2,201–2,300 of 4,946, ranked by callers