JAX
There is a growing number of startups exploring alternative compute paradigms; i.e. companies not making GPUs/CPUs/TPUs (such as Mythic, Vaire, Ludwig Computing, Extropic, Normal Computing, Unconventional, and many more). Each company is approaching the problem of computing in slightly different ways, and in this blog we will
JAX
Originally Publish: February 11, 2026 Note: this was published before hijax became somewhat established in JAX, I will investigate the new changes and put out a new blog on them. Stateful operations in JAX, such as batchnorm, can sometimes be annoying to work with. As a DSL that relies on
JAX
Originally published: December 13, 2025 In this blog, we'll be looking at one of the "sharp bits" of JAX: pseudorandomness, and see if we can't smooth it out a little. JAX is an accelerated linear algebra library that forms the backbone of some modern