jax

afa3c0a4 - [pallas:triton] Fix atomic min/max lowering for unsigned integers and float types (#264)