Select speed comparisons with Pytorch and {NumPy, Tensorflow}
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
images
.gitignore
Grad-timings.ipynb
NumPy and PyTorch speed comparison.ipynb
README.md
Tensorflow-eval.ipynb
grad-timings.csv
speed_tests.py

README.md

Comparing with NumPy gradient calculation:

Comparing with many different NumPy functions:

Comaring torch.numpy() with tf.session.eval:

...but that's not a super useful visualization. Let's use a log y-axis instead: