George Necula fd52fbf411 Fix import in benchmarks ...

This works on my machine as 'python benchmarks/pmap_benchmark.py'. It also
follows the code in examples.

This will need a copybara rule to change the import to 'from jax.benchmarks import benchmark'

2020-03-31 11:48:08 +03:00

..

benchmark.py

Added type annotations and removed unused imports (#2472)

2020-03-21 13:54:30 +01:00

pmap_benchmark.py

Fix import in benchmarks

2020-03-31 11:48:08 +03:00