Repository navigation

Tpu backend - #713

Closed

drbh wants to merge 13 commits into

mainfrom

Closed

Tpu backend#713
drbh wants to merge 13 commits into
mainfrom
tpu

Commits on Jul 13, 2026
- kernels: add support for TPU
  tengomucho
  committed
  9bd1937
- kernels-data: add Backend::Tpu enum variant
  tengomucho
  committed
  a780b4b
- kernel-builder: detect TPU in noarch _ops.py template
  tengomucho
  committed
  dac23cd
- examples: add kernels-test/relu-tpu example kernel repository
  tengomucho
  committed
  31c394b
- kernels/pytest: register tpu_only marker
  tengomucho
  committed
  36a9d72
- nix-builder: add TPU backend support
  tengomucho
  committed
  4359655
- nix-builder: add TPU build configuration to versions.nix
  
  tengomucho
  and
  claude
  committed
  ef27628
- examples: add flake.nix to relu-tpu kernel
  
  tengomucho
  and
  claude
  committed
  38712a1
- nix-builder: build TPU wheels against the default python
  
  tengomucho
  and
  claude
  committed
  ee10f69
- kernels-data,kernels: resolve TPU python deps to nix packages
  
  tengomucho
  and
  claude
  committed
  3cd1ac2
- kernels,kernel-builder: detect TPU via torch.backends.tpu
  
  tengomucho
  and
  claude
  committed
  cae1a7d
- kernels-data: include TPU python-depends in backend metadata
  
  tengomucho
  and
  claude
  committed
  4bf3946
- examples/relu-tpu: annotate the jax function signature
  
  tengomucho
  and
  claude
  committed
  7126d26