3 ms·
The default compiler is based on Google XLA, which is the same as used by JAX, so expect similar numbers to said Python library. :)
by josevalim 4y ago
The default compiler is based on Google XLA, which is the same as used by JAX, so expect similar numbers to said Python library. :)