Jake VanderPlas 368d3433a6 Add random benchmarks ...

The purpose of this is to measure the difference in dispatch seed between raw keys and new-style typed keys. The latter does not yet hit the C++ fast path, and so we expect it to incur a small additional overhead at dispatch time. Part of #9263.

PiperOrigin-RevId: 559782442

2023-08-24 09:55:07 -07:00

..

api_benchmark.py

[jax][benchmark] Added clearing caches for benchmarking compilation time in sparse JAX benchmarks

2023-08-02 10:07:54 -07:00

math_benchmark.py

Update flags to use the ABSL typed flag API.

2023-07-27 12:15:58 -07:00

random_benchmark.py

Add random benchmarks

2023-08-24 09:55:07 -07:00

sparse_benchmark.py

[jax][benchmark] Added clearing caches for benchmarking compilation time in sparse JAX benchmarks

2023-08-02 10:07:54 -07:00