Yash Katariya 0b542ff585 Add a benchmark measuring device_put's speed for a 4GB input array ...

```
---------------------------------------------------------
Benchmark               Time             CPU   Iterations
---------------------------------------------------------
device_put_big        419 ms        0.363 ms           10
```

PiperOrigin-RevId: 607512568

2024-02-15 17:47:10 -08:00

..

api_benchmark.py

Add a benchmark measuring device_put's speed for a 4GB input array

2024-02-15 17:47:10 -08:00

linalg_benchmark.py

Use a Jacobi SVD solver for unbatched SVDs up to 1024x1024 on NVIDIA GPUs.

2023-11-13 12:04:13 -08:00

math_benchmark.py

Update flags to use the ABSL typed flag API.

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

random_benchmark.py

fast dispatch for functions over typed PRNG key arrays

2023-09-13 09:43:58 -07:00

shape_poly_benchmark.py

[shape_poly] Add another micro-benchmark

2024-02-09 09:02:59 -08: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