Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ml-explore/mlx
/ functions
Functions
4,946 in github.com/ml-explore/mlx
⨍
Functions
4,946
◇
Types & classes
770
↳
Endpoints
5
Method
test_save_npz_weights
(self)
python/tests/test_nn.py:88
Method
test_save_safetensors_weights
(self)
python/tests/test_nn.py:106
Method
test_savez_and_loadz
(self)
python/tests/test_load.py:341
Method
test_scalar_conversion
(self)
python/tests/test_array.py:66
Method
test_scalar_inputs
(self)
python/tests/test_ops.py:95
Method
test_scalar_integer_conversion_overflow
(self)
python/tests/test_array.py:2244
Method
test_scans
(self)
python/tests/test_ops.py:2068
Method
test_scatter_add_vjp
(self)
python/tests/test_autograd.py:317
Method
test_scatter_max_vjp
(self)
python/tests/test_autograd.py:330
Method
test_scatter_min_vjp
(self)
python/tests/test_autograd.py:350
Method
test_scatter_vjp
(self)
python/tests/test_autograd.py:298
Method
test_schedule_joiner
(self)
python/tests/test_optimizers.py:449
Method
test_sdpa
(self)
python/tests/test_fast_sdpa.py:364
Method
test_sdpa_attention_sinks
(self)
python/tests/test_fast_sdpa.py:556
Method
test_sdpa_broadcast_mask
(self)
python/tests/test_fast_sdpa.py:485
Method
test_sdpa_few_query
(self)
python/tests/test_fast_sdpa.py:248
Method
test_sdpa_fully_masked
(self)
python/tests/test_fast_sdpa.py:224
Method
test_sdpa_grad
(self)
python/tests/test_fast_sdpa.py:606
Method
test_sdpa_inf_score
(self)
python/tests/test_fast_sdpa.py:236
Method
test_sdpa_long_masked_sequence
(self)
python/tests/test_fast_sdpa.py:448
Method
test_sdpa_nan_bug
(self)
python/tests/test_fast_sdpa.py:528
Method
test_sdpa_noncontiguous_inputs
(self)
python/tests/test_fast_sdpa.py:501
Method
test_sdpa_promote_mask
(self)
python/tests/test_fast_sdpa.py:512
Method
test_sdpa_sliced
(self)
python/tests/test_fast_sdpa.py:653
Method
test_sdpa_vector
(self)
python/tests/test_fast_sdpa.py:152
Method
test_sdpa_vector_batched
(self)
python/tests/test_fast_sdpa.py:323
Method
test_sdpa_vector_kv_transposed_head_seq
(self)
python/tests/test_fast_sdpa.py:120
Method
test_sdpa_vector_value_dims
(self)
python/tests/test_fast_sdpa.py:307
Method
test_segmented_mm
(self)
python/tests/test_blas.py:1294
Method
test_send_recv
(self)
python/tests/mpi_test_distributed.py:115
Method
test_send_recv
(self)
python/tests/ring_test_distributed.py:80
Method
test_sequential
(self)
python/tests/test_nn.py:878
Method
test_set_dtype
(self)
python/tests/test_nn.py:1877
Method
test_setitem
(self)
python/tests/test_array.py:1137
Method
test_setitem_with_boolean_mask
(self)
python/tests/test_array.py:2077
Method
test_setitem_with_list
(self)
python/tests/test_array.py:2036
Method
test_sgd
(self)
python/tests/test_optimizers.py:77
Method
test_shape
( M, N, K, dtype=mx.float32, batch_A=(),
python/tests/test_quantized.py:768
Method
test_shape
( M, N, K, block_size, transpose=False,
python/tests/test_blas.py:924
Method
test_shapeless_compile
(self)
python/tests/test_compile.py:399
Method
test_shapeless_compile_addmm
(self)
python/tests/test_compile.py:887
Method
test_shapeless_compile_full_like
(self)
python/tests/test_compile.py:485
Method
test_shapeless_compile_gather
(self)
python/tests/test_compile.py:477
Method
test_shapeless_compile_matmul
(self)
python/tests/test_compile.py:880
Method
test_shapeless_compile_slice_update
(self)
python/tests/test_compile.py:915
Method
test_shapeless_compile_unflatten
(self)
python/tests/test_compile.py:469
Method
test_shapeless_compile_with_broadcasts
(self)
python/tests/test_compile.py:423
Method
test_shapeless_compile_with_reduction
(self)
python/tests/test_compile.py:437
Method
test_shapeless_compile_with_reshape
(self)
python/tests/test_compile.py:928
Method
test_shapeless_mean
(self)
python/tests/test_compile.py:700
Method
test_shard_linear
(self)
python/tests/mlx_distributed_tests.py:113
Method
test_shard_predicate
(self)
python/tests/mlx_distributed_tests.py:266
Method
test_shared_broadcast
(self)
python/tests/test_compile.py:1058
Method
test_siblings_without_eval
(self)
python/tests/test_array.py:2222
Method
test_sigmoid
(self)
python/tests/test_nn.py:935
Method
test_sigmoid
(self)
python/tests/test_ops.py:1071
Method
test_sign
(self)
python/tests/test_ops.py:853
Method
test_simple_compile
(self)
python/tests/test_compile.py:16
Method
test_simple_einsum
(self)
python/tests/test_einsum.py:68
Method
test_simple_path
(self)
python/tests/test_einsum.py:12
Method
test_sin
(self)
python/tests/test_ops.py:1023
Method
test_sin_pe
(self)
python/tests/test_nn.py:924
Method
test_slice_grads
(self)
python/tests/test_autograd.py:795
Method
test_slice_into_layer_norm
(self)
python/tests/test_fast.py:588
Method
test_slice_negative_step
(self)
python/tests/test_array.py:1550
Method
test_slice_update_add_vjp
(self)
python/tests/test_autograd.py:410
Method
test_slice_update_max_vjp
(self)
python/tests/test_autograd.py:370
Method
test_slice_update_min_vjp
(self)
python/tests/test_autograd.py:390
Method
test_slice_update_multiply_vjp
(self)
python/tests/test_autograd.py:423
Method
test_slice_update_reversed
(self)
python/tests/test_ops.py:3139
Method
test_slice_with_negative_stride
(self)
python/tests/test_ops.py:3145
Method
test_small_matrix
(self)
python/tests/test_quantized.py:587
Method
test_smooth_l1_loss
(self)
python/tests/test_losses.py:217
Method
test_softmax
(self)
python/tests/test_nn.py:986
Method
test_softmax
(self)
python/tests/test_ops.py:1800
Method
test_softmin
(self)
python/tests/test_nn.py:995
Method
test_softplus
(self)
python/tests/test_nn.py:1004
Method
test_softshrink
(self)
python/tests/test_nn.py:1022
Method
test_softsign
(self)
python/tests/test_nn.py:1013
Method
test_solve
(self)
python/tests/test_linalg.py:540
Method
test_solve_triangular
(self)
python/tests/test_linalg.py:594
Method
test_sort
(self)
python/tests/test_ops.py:2193
Method
test_sort_nan
(self)
python/tests/test_ops.py:3329
Method
test_sparse
(self)
python/tests/test_init.py:92
Method
test_split
(self)
python/tests/test_ops.py:1310
Method
test_split_against_slice
(self)
python/tests/test_autograd.py:436
Method
test_split_invalid_num_splits
Regression: split with num_splits <= 0 should raise, not crash.
python/tests/test_ops.py:1331
Method
test_sqrt
(self)
python/tests/test_ops.py:903
Method
test_square
(self)
python/tests/test_ops.py:896
Method
test_squeeze_expand
(self)
python/tests/test_ops.py:2177
Method
test_stack
(self)
python/tests/test_ops.py:2378
Method
test_state
(x)
python/tests/test_compile.py:314
Method
test_std
(self)
python/tests/test_ops.py:834
Method
test_step_decay
(self)
python/tests/test_optimizers.py:424
Method
test_stop_gradient
(self)
python/tests/test_autograd.py:259
Method
test_stop_gradient
(self)
python/tests/test_ops.py:1132
Method
test_stream
(self)
python/tests/test_device.py:68
Method
test_subtract
(self)
python/tests/test_ops.py:234
Method
test_success
()
python/tests/test_threads.py:30
Method
test_sum
(self)
python/tests/test_ops.py:607
← previous
next →
4,601–4,700 of 4,946, ranked by callers