mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-25 09:51:17 +08:00
1b97b2958b
* Simple kernel generation * Remove the generate kernel from graph_utils * fix multi-output with compile * fuse with stopgrad * v1 input, output capture in compile * cleanup tree update with visitor update * nit * remove todo * state for model, optional explicit init and more pure optimizer steps * move learning rate to state * add lr to opt state, some fixes in capture * fix optim * update tuple of containers as well * fix stream for compiled output * rng state for compile * nit * updates and comments --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com> |
||
|---|---|---|
| .. | ||
| mlx_tests.py | ||
| test_array.py | ||
| test_autograd.py | ||
| test_bf16.py | ||
| test_blas.py | ||
| test_compile.py | ||
| test_constants.py | ||
| test_conv.py | ||
| test_device.py | ||
| test_eval.py | ||
| test_fft.py | ||
| test_graph.py | ||
| test_init.py | ||
| test_linalg.py | ||
| test_load.py | ||
| test_losses.py | ||
| test_nn.py | ||
| test_ops.py | ||
| test_optimizers.py | ||
| test_quantized.py | ||
| test_random.py | ||
| test_reduce.py | ||
| test_tree.py | ||
| test_vmap.py | ||