v0.3.2
What's Changed
- [pallas-tpu] Enable test_geglu, test_swiglu on Pallas; fix softmax/kl_div/gdn xfails by @v0i0 in #1637
- [pallas-tpu] make dot work: use jnp.matmul and f32 accumulator by @v0i0 in #1647
- [pallas-tpu] Enable broadcasting and control flow tests for Pallas backend by @v0i0 in #1643
- Use ephemeral Triton cache during autotuning to eliminate loser artifacts by @fulvius31 in #1649
- Try public bazel cache for torch_tpu build by @v0i0 in #1658
- fix cudagraph grouped gemm fail in benchmarks by @v0i0 in #1662
- Use matmul fwd direclty in autograd for performance by @tianrengao in #1045
- Recreate foreign SymInts in helion's own ShapeEnv by @gmagogsfm in #1671
Full Changelog: v0.3.1...v0.3.2