Skip to content

Commit 721b517

Browse files
justinjfuGoogle-ML-Automation
authored andcommitted
[Pallas] Update changelog for pl.estimate_cost
PiperOrigin-RevId: 702767883
1 parent 653f654 commit 721b517

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

docs/pallas/CHANGELOG.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,11 @@ Remember to align the itemized text with the first line of an item within a list
2020
{func}`jax.experimental.tpu.run_scoped`. Both are now available in
2121
{mod}`jax.experimental.pallas`.
2222

23+
* New functionality
24+
25+
* Added a cost estimate tool {func}`pl.estimate_cost` for automatically
26+
constructing a kernel cost estimate from a JAX reference function.
27+
2328
## Released with jax 0.4.34 (October 4, 2024)
2429

2530
* Changes

0 commit comments

Comments
 (0)