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
Methodtest_sum_bool
(self)
python/tests/test_reduce.py:127
Methodtest_sum_first
(self)
python/tests/test_einsum.py:180
Methodtest_sum_scatter
(self)
python/tests/nccl_test_distributed.py:17
Methodtest_supported_trees
(self)
python/tests/test_tree.py:49
Methodtest_svd_decomposition
(self)
python/tests/test_linalg.py:137
Methodtest_take
(self)
python/tests/test_ops.py:1169
Methodtest_take_along_axis
(self)
python/tests/test_ops.py:1253
Methodtest_tensordot
(self)
python/tests/test_ops.py:2517
Methodtest_threadlocal_stream
(self)
python/tests/test_threads.py:11
Methodtest_throw
(self)
python/tests/test_quantized.py:571
Methodtest_tile
(self)
python/tests/test_ops.py:2597
Methodtest_to_dot
(self)
python/tests/test_graph.py:11
Methodtest_to_from_fp8
(self)
python/tests/test_ops.py:3349
Methodtest_to_scalar
(self)
python/tests/test_array.py:2166
Methodtest_topk_grad
(self)
python/tests/test_autograd.py:593
Methodtest_torch_conv_1D
(self)
python/tests/test_conv.py:59
Methodtest_torch_conv_1D_grad
(self)
python/tests/test_conv.py:168
Methodtest_torch_conv_2D
(self)
python/tests/test_conv.py:274
Methodtest_torch_conv_2D_grad
(self)
python/tests/test_conv.py:374
Methodtest_torch_conv_3D
(self)
python/tests/test_conv.py:484
Methodtest_torch_conv_3D_grad
(self)
python/tests/test_conv.py:568
Methodtest_torch_conv_depthwise
(self)
python/tests/test_conv.py:1056
Methodtest_torch_conv_general
(self)
python/tests/test_conv.py:794
Methodtest_torch_conv_tranpose_1d_output_padding
(self)
python/tests/test_conv_transpose.py:600
Methodtest_torch_conv_transpose_1D
(self)
python/tests/test_conv_transpose.py:22
Methodtest_torch_conv_transpose_1D_grad
(self)
python/tests/test_conv_transpose.py:131
Methodtest_torch_conv_transpose_2D
(self)
python/tests/test_conv_transpose.py:229
Methodtest_torch_conv_transpose_2D_grad
(self)
python/tests/test_conv_transpose.py:319
Methodtest_torch_conv_transpose_2d_output_padding
(self)
python/tests/test_conv_transpose.py:656
Methodtest_torch_conv_transpose_3D
(self)
python/tests/test_conv_transpose.py:416
Methodtest_torch_conv_transpose_3D_grad
(self)
python/tests/test_conv_transpose.py:494
Methodtest_torch_conv_transpose_3d_output_padding
(self)
python/tests/test_conv_transpose.py:731
Methodtest_torch_upsample
(self)
python/tests/test_upsample.py:21
Methodtest_trace
(self)
python/tests/test_ops.py:2680
Methodtest_transform_over_eval_compiled
(self)
python/tests/test_compile.py:288
Methodtest_transformer
(self)
python/tests/test_nn.py:2072
Methodtest_transformer_decoder
(self)
python/tests/test_nn.py:2065
Methodtest_transformer_decoder_layer
(self)
python/tests/test_nn.py:2031
Methodtest_transformer_encoder
(self)
python/tests/test_nn.py:2059
Methodtest_transformer_encoder_layer
(self)
python/tests/test_nn.py:2009
Methodtest_transpose_axis
(self)
python/tests/test_ops.py:586
Methodtest_transpose_noargs
(self)
python/tests/test_ops.py:575
Methodtest_tree
(self)
python/tests/test_vmap.py:111
Methodtest_tree_flatten
(self)
python/tests/test_tree.py:19
Methodtest_tree_map
(self)
python/tests/test_tree.py:12
Methodtest_tri
(self)
python/tests/test_ops.py:431
Methodtest_tri_inverse
(self)
python/tests/test_linalg.py:221
Methodtest_trig_ops
(self)
python/tests/test_ops.py:1587
Methodtest_tril
(self)
python/tests/test_ops.py:438
Methodtest_triplet_loss
(self)
python/tests/test_losses.py:337
Methodtest_triu
(self)
python/tests/test_ops.py:445
Methodtest_truncated_normal
(self)
python/tests/test_random.py:260
Methodtest_tuple_equals_array
(self)
python/tests/test_array.py:152
Methodtest_tuple_not_equals_array
(self)
python/tests/test_array.py:207
Methodtest_two_input_einsum
(self)
python/tests/test_einsum.py:136
Methodtest_type_promotion
(self)
python/tests/test_double.py:176
Methodtest_types_conserved
(self)
python/tests/test_optimizers.py:69
Methodtest_unary
(self)
python/tests/test_vmap.py:33
Methodtest_unary_ops
(self)
python/tests/test_double.py:13
Methodtest_unary_ops
(self)
python/tests/test_bf16.py:107
Methodtest_unary_ops
(self)
python/tests/test_ops.py:1520
Methodtest_unary_ops_from_non_array
(self)
python/tests/test_ops.py:1538
Methodtest_uniform
(self)
python/tests/test_init.py:33
Methodtest_uniform
(self)
python/tests/test_random.py:44
Methodtest_unsupported_input_types
(self)
python/tests/test_compile.py:638
Methodtest_update
(self)
python/tests/test_nn.py:278
Methodtest_update_lr_compiled
(self)
python/tests/test_optimizers.py:394
Methodtest_update_modules
(self)
python/tests/test_nn.py:295
Methodtest_update_state
(self)
python/tests/test_autograd.py:283
Methodtest_upsample
(self)
python/tests/test_nn.py:1193
Methodtest_var
(self)
python/tests/test_ops.py:811
Methodtest_version
(self)
python/tests/test_array.py:28
Methodtest_view
(self)
python/tests/test_ops.py:2901
Methodtest_vjp
(self)
python/tests/test_autograd.py:51
Methodtest_vjp
(slow, fast, primals)
python/tests/test_fast_sdpa.py:610
Methodtest_vjp_scales_biases
(self)
python/tests/test_quantized.py:1107
Methodtest_vjp_types
(self)
python/tests/test_autograd.py:455
Methodtest_vjp_vjp_compiled
(self)
python/tests/test_compile.py:252
Methodtest_vmap_argreduce
(self)
python/tests/test_vmap.py:245
Methodtest_vmap_compiled
(self)
python/tests/test_compile.py:210
Methodtest_vmap_concatenate
(self)
python/tests/test_vmap.py:557
Methodtest_vmap_const_func
(self)
python/tests/test_vmap.py:537
Methodtest_vmap_conv
(self)
python/tests/test_vmap.py:681
Methodtest_vmap_flatten
(self)
python/tests/test_vmap.py:671
Methodtest_vmap_gather
(self)
python/tests/test_vmap.py:400
Methodtest_vmap_indexing
(self)
python/tests/test_vmap.py:167
Methodtest_vmap_inverse
(self)
python/tests/test_vmap.py:370
Methodtest_vmap_masked_scatter
(self)
python/tests/test_vmap.py:815
Methodtest_vmap_matmul
(self)
python/tests/test_vmap.py:277
Methodtest_vmap_mean
(self)
python/tests/test_vmap.py:255
Methodtest_vmap_pad
(self)
python/tests/test_vmap.py:732
Methodtest_vmap_put_along_axis
(self)
python/tests/test_vmap.py:604
Methodtest_vmap_reduce
(self)
python/tests/test_vmap.py:222
Methodtest_vmap_scatter
(self)
python/tests/test_vmap.py:445
Methodtest_vmap_split_vmap
(self)
python/tests/test_vmap.py:630
Methodtest_vmap_svd
(self)
python/tests/test_vmap.py:316
Methodtest_vmap_take_along_axis
(self)
python/tests/test_vmap.py:577
Methodtest_vmap_types
(self)
python/tests/test_vmap.py:770
Methodtest_where
(self)
python/tests/test_double.py:98
Methodtest_where
(self)
python/tests/test_ops.py:1980
← previousnext →4,701–4,800 of 4,946, ranked by callers