Jake VanderPlas 5dc37d3f70 Remove internal uses of api_util.shaped_abstractify

2024-12-19 07:06:36 -08:00

..

mosaic

Replace disable_backends with enable_backends on jax_multiplatform_test.

2024-09-27 06:15:31 -07:00

api_benchmark.py

Remove internal uses of api_util.shaped_abstractify

2024-12-19 07:06:36 -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] Adding shape polymorphism support for the state primitives.

2024-11-21 06:17:01 -08:00

sparse_benchmark.py

update README and several docs to typed RNG keys

2024-08-11 08:09:47 -07:00