MCPcopy Create free account
hub / github.com/ml-explore/mlx / seed

Method seed

python/src/random.cpp:32–34  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

30 }
31
32 void seed(uint64_t seed) {
33 state()[0] = mx::random::key(seed);
34 }
35
36 mx::array next() {
37 auto out = mx::random::split(nb::cast<mx::array>(state()[0]));

Callers 15

init_randomFunction · 0.45
run_conv1DMethod · 0.45
run_conv1D_gradMethod · 0.45
run_conv2DMethod · 0.45
run_conv2D_gradMethod · 0.45
run_conv3DMethod · 0.45
run_conv3D_gradMethod · 0.45
__conv_general_testMethod · 0.45
test_conv_groups_gradMethod · 0.45
test_lapackMethod · 0.45
test_rope_with_freqsMethod · 0.45

Calls 1

keyFunction · 0.85

Tested by 15

run_conv1DMethod · 0.36
run_conv1D_gradMethod · 0.36
run_conv2DMethod · 0.36
run_conv2D_gradMethod · 0.36
run_conv3DMethod · 0.36
run_conv3D_gradMethod · 0.36
__conv_general_testMethod · 0.36
test_conv_groups_gradMethod · 0.36
test_lapackMethod · 0.36
test_rope_with_freqsMethod · 0.36