• Turun@feddit.de
    link
    fedilink
    arrow-up
    7
    ·
    edit-2
    1 year ago

    I personally did one project with PyO3 and it was a breeze.

    For data analysis I initially implemented my physics model in python (say f=m*a), but during a fitting procedure (what is “a”, given a measured “m” and “f”?) this part of the code is called thousands of times.

    Writing such a simple function is rust (twi numbers in (m and a guess for a), one number out (predicted f)) was easy. Since there are no complex data structures involved the borrow checker was happy the whole way through. Rust has bindings for numpy with the ndarray crate/library, so even that was simple.

    Simply writing the numerics in rust gave a 40x speedup in my case. I got another 2.5x by making the main loop of my calculation parallel. With the rayon crate this is again a single line change, turning for x in array {...} into for x in array.par_iter() {...}

    I can’t recommend it enough. If you have a single hot spot in your python code, this is the way to go, even if you are new to rust.