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

Added stft and istft in the header

Browse Source
This commit is contained in:
paramthakkar123
2025-04-23 19:19:04 +05:30
parent a963a15b8d
commit f7f323f6ae
4 changed files with 56 additions and 10 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));
}