[Open Source] Fused Gated DeltaNet-2 training kernels for TPU v5e in JAX/Pallas
This story is from 2026-09-11. It is preserved in the archive; the latest stories are on the live feed.
I released an independent open-source implementation of fused Gated DeltaNet-2 (GDN-2) training kernels for TPU v5e using JAX/Pallas. Repository: https://github.com/Akseleu-J/atomic-ops Citable release: https://doi.org/10.5281/zenodo.22706659 The repository contains three implementations of the sam…
Read the full story at r/deeplearning ↗
Timeline · 2 reports
- 2026-09-11 12:42 · r/machinelearningnews
Open-source fused Gated DeltaNet-2 training kernels for TPU v5e in JAX/Pallas - 2026-09-11 12:26 · r/deeplearning
[Open Source] Fused Gated DeltaNet-2 training kernels for TPU v5e in JAX/Pallas