Files
mlx/tests/vmap_tests.cpp
Angelos Katharopoulos a611b0bc82 Removes the retain_graph flag (#385)
* Adds global tracing flag
* Removes retain_graph in favor of is_tracer
2024-01-07 15:16:51 -08:00

11 KiB