Closed FHof closed 2 years ago
This makes MonteCarlo faster with torch, tensorflow and uncompiled jax, especially when the dimensionality is high.
Python for loop (before):
Broadcasting (after):
This makes MonteCarlo faster with torch, tensorflow and uncompiled jax, especially when the dimensionality is high.
Python for loop (before):
Broadcasting (after):