Zac Mustin 0c8e601f90 Support convolution in roofline. ...

So far we support only `unfused_hmb_bytes` and don't account for `{feature, batch}_group_count`s due to complexity.

PiperOrigin-RevId: 736948528

2025-03-14 12:26:20 -07:00

..

__init__.py

Add experimental JAX roofline API.

2024-11-27 14:38:57 -08:00

roofline.py

Add a variant of safe_map() that has no return value, named foreach().

2025-03-14 07:42:48 -07:00

rooflines.py

Support convolution in roofline.

2025-03-14 12:26:20 -07:00