3 ms·
Does Bumblebee has any significant performance improvements over the python implementation?
by hargup 4y ago
Does Bumblebee has any significant performance improvements over the python implementation?
- itake 4y agoI am guessing no, otherwise that would have been showcased. I think most of the python implementations are actually C/C++ implementations, which would be hard to beat, or would be easily patched to be faster.
- xrd 4y agoI linked to a podcast (but here it is again) where Sean discussed the implications of using XLA and how that reduces the translation back and forth between python and the numerical libraries underneath. It sounds like the Axon team made some choices that are paying off in the in that respect and that other teams are migrating towards. https://podcast.thinkingelixir.com/102 https://podcast.thinkingelixir.com/102
- josevalim 4y agoThe default compiler is based on Google XLA, which is the same as used by JAX, so expect similar numbers to said Python library. :)