We are an independent lab working on demystifying machine learning accelerators.

Projects

• TPU trace viewer
  Send a JAX program to a simulated TPU slice and visualise a trace of the simulated run.

Writing

• Simulating Google TPUs
  A simulator that runs Google's real TPU stack, how close it gets to the chip, and what we learned building it.

Reach us at contact@gherkinlabs.com.