zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Merge eeaf1fa463 into 4ad53414dd

Browse Source
This commit is contained in:
Param Thakkar
2025-07-26 19:00:49 +09:00
committed by GitHub
parent 4ad53414dd eeaf1fa463
commit d7e81cd331
5 changed files with 338 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
benchmarks/cpp/autograd.cpp
View File

@@ -10,7 +10,7 @@ namespace mx = mlx::core;
void time_value_and_grad() {
auto x = mx::ones({200, 1000});
mx::eval(x);
auto fn = [](mx::array x) {
auto fn = [](mx::x) {
for (int i = 0; i < 20; ++i) {
x = mx::log(mx::exp(x));
}