Software Engineer @ Google TPU Working on XLA compiler
pyptx. A Python DSL to write Nvidia PTX for Hopper and Blackwell in JAX and PyTorch
justabyte. Code snippets and reproductions from JustAByte
xla. A machine learning compiler for GPUs, CPUs, and ML accelerators
axlearn. Python
