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_compile_types
(self)
python/tests/test_compile.py:1232
Methodtest_compile_vjp
(self)
python/tests/test_compile.py:660
Methodtest_compile_with_closure
(self)
python/tests/test_compile.py:105
Methodtest_compile_with_constant
(self)
python/tests/test_compile.py:507
Methodtest_compile_with_long_name
(self)
python/tests/test_compile.py:732
Methodtest_compile_with_none
(self)
python/tests/test_compile.py:1124
Methodtest_compile_with_schedule
(self)
python/tests/test_optimizers.py:480
Methodtest_compile_without_captured_inputs
(self)
python/tests/test_compile.py:788
Methodtest_compiled_optimizer
(self)
python/tests/test_optimizers.py:336
Methodtest_compiled_preserves_attributes
(self)
python/tests/test_compile.py:1111
Methodtest_complex_gemm
(self)
python/tests/test_blas.py:1398
Methodtest_complex_gemv
(self)
python/tests/test_blas.py:1368
Methodtest_complex_norm
(self)
python/tests/test_linalg.py:68
Methodtest_complex_normal
(self)
python/tests/test_random.py:355
Methodtest_complex_ops
(self)
python/tests/test_ops.py:3155
Methodtest_complex_power
(self)
python/tests/test_ops.py:3184
Methodtest_complex_prod_vjp
(self)
python/tests/test_autograd.py:969
Methodtest_complex_vjps
(self)
python/tests/test_autograd.py:687
Methodtest_concatenate
(self)
python/tests/test_ops.py:1850
Methodtest_concatenate_vjps
(self)
python/tests/test_autograd.py:717
Methodtest_conjugate
(self)
python/tests/test_ops.py:2878
Methodtest_constant
(self)
python/tests/test_init.py:11
Methodtest_constants_availability
(self)
python/tests/test_constants.py:24
Methodtest_constants_values
(self)
python/tests/test_constants.py:11
Methodtest_construction_from_lists
(self)
python/tests/test_array.py:377
Methodtest_construction_from_lists_of_mlx_arrays
(self)
python/tests/test_array.py:441
Methodtest_conv1d
(self)
python/tests/test_nn.py:788
Methodtest_conv2d
(self)
python/tests/test_nn.py:824
Methodtest_conv2d_large_filter_small_channels
(self)
python/tests/test_conv.py:1190
Methodtest_conv2d_unaligned_channels
(self)
python/tests/test_conv.py:1177
Methodtest_conv_1d_groups_flipped
(self)
python/tests/test_conv.py:51
Methodtest_conv_1d_with_2d
(self)
python/tests/test_conv.py:1156
Methodtest_conv_general_flip_grad
(self)
python/tests/test_conv.py:885
Methodtest_conv_groups_grad
(self)
python/tests/test_conv.py:954
Methodtest_conversion
(self)
python/tests/test_double.py:290
Methodtest_conversion
(self)
python/tests/test_bf16.py:194
Methodtest_conversion_ml_dtypes
(self)
python/tests/test_bf16.py:202
Methodtest_cos
(self)
python/tests/test_ops.py:1032
Methodtest_cosine_decay
(self)
python/tests/test_optimizers.py:436
Methodtest_cosine_similarity_loss
(self)
python/tests/test_losses.py:381
Methodtest_create_from_buffer
(self)
python/tests/test_array.py:1805
Methodtest_cross_entropy
(self)
python/tests/test_losses.py:12
Methodtest_cross_product
(self)
python/tests/test_linalg.py:303
Methodtest_cumprod_grad
(self)
python/tests/test_autograd.py:520
Methodtest_custom_function
(self)
python/tests/test_autograd.py:603
Methodtest_custom_kernel_args
(self)
python/tests/test_fast.py:773
Methodtest_custom_kernel_attributes
(self)
python/tests/test_fast.py:932
Methodtest_custom_kernel_basic
(self)
python/tests/test_fast.py:740
Methodtest_custom_kernel_caching
(self)
python/tests/test_fast.py:958
Methodtest_custom_kernel_helper
(self)
python/tests/test_fast.py:886
Methodtest_custom_kernel_strides
(self)
python/tests/test_fast.py:832
Methodtest_decay_lr
(self)
python/tests/test_optimizers.py:411
Methodtest_deep_graphs
(self)
python/tests/test_array.py:2194
Methodtest_degrees
(self)
python/tests/test_ops.py:1041
Methodtest_depends
(self)
python/tests/test_ops.py:3217
Methodtest_device
(self)
python/tests/test_device.py:25
Methodtest_device_context
(self)
python/tests/test_device.py:42
Methodtest_device_count
(self)
python/tests/test_device.py:117
Methodtest_device_info_cpu
(self)
python/tests/test_device.py:126
Methodtest_device_info_default
(self)
python/tests/test_device.py:143
Methodtest_device_info_gpu
(self)
python/tests/test_device.py:134
Methodtest_diag
(self)
python/tests/test_ops.py:2638
Methodtest_diagonal
(self)
python/tests/test_ops.py:2624
Methodtest_divide
(self)
python/tests/test_ops.py:266
Methodtest_divmod
(self)
python/tests/test_ops.py:2575
Methodtest_dlpack
(self)
python/tests/test_array.py:2001
Methodtest_dlx_device_type
(self)
python/tests/test_array.py:194
Methodtest_donation
(self)
python/tests/mlx_distributed_tests.py:95
Methodtest_donation_for_noops
(self)
python/tests/test_eval.py:139
Methodtest_double_constant
(self)
python/tests/test_compile.py:1047
Methodtest_double_keeps_precision
(self)
python/tests/test_array.py:433
Methodtest_dropout
(self)
python/tests/test_nn.py:1145
Methodtest_dropout2d
(self)
python/tests/test_nn.py:1161
Methodtest_dropout3d
(self)
python/tests/test_nn.py:1177
Methodtest_dtype_promotion
(self)
python/tests/test_array.py:776
Methodtest_dtype_python_scalar_promotion
(self)
python/tests/test_array.py:817
Methodtest_dtypes
(self)
python/tests/test_reduce.py:50
Methodtest_dtypes
(self)
python/tests/test_compile.py:774
Methodtest_dtypes
(self)
python/tests/test_array.py:37
Methodtest_dynamic_slicing
(self)
python/tests/test_ops.py:3113
Methodtest_edge_case
(self)
python/tests/test_reduce.py:118
Methodtest_eig
(self)
python/tests/test_linalg.py:351
Methodtest_eigh
(self)
python/tests/test_linalg.py:435
Methodtest_ellipses
(self)
python/tests/test_einsum.py:316
Methodtest_elu
(self)
python/tests/test_nn.py:963
Methodtest_empty_matmul
(self)
python/tests/test_blas.py:771
Methodtest_empty_matmuls
(self)
python/tests/test_ops.py:2614
Methodtest_enable_disable
(self)
python/tests/test_compile.py:172
Methodtest_erf
(self)
python/tests/test_ops.py:995
Methodtest_erfinv
(self)
python/tests/test_ops.py:1001
Methodtest_eval
(self)
python/tests/test_eval.py:11
Methodtest_eval_in_grad
(self)
python/tests/test_autograd.py:490
Methodtest_eval_mixed
(self)
python/tests/test_eval.py:27
Methodtest_eval_slow_fast_multi_stream
(self)
python/tests/test_eval.py:92
Methodtest_exp
(self)
python/tests/test_ops.py:980
Methodtest_expand_sums
(self)
python/tests/test_reduce.py:29
Methodtest_expm1
(self)
python/tests/test_ops.py:987
Methodtest_exponential_decay
(self)
python/tests/test_optimizers.py:430
Methodtest_export_control_flow
(self)
python/tests/test_export_import.py:448
Methodtest_export_conv
(self)
python/tests/test_export_import.py:316
← previousnext →4,201–4,300 of 4,946, ranked by callers