This is a simple benchmark of a transformer language model implemented in JAX and PyTorch.
See blog post: http://www.echonolan.net/posts/2021-09-06-JAX-vs-PyTorch-A-Transformer-Benchmark.html
-
Notifications
You must be signed in to change notification settings - Fork 0
enolan/jax-torch-bench
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published