Trainium expert
us-east-2c
hardware: AWS Trainium trn1
stack: PyTorch / torch_xla
trained on: digits 0-4
CPU router
europe-west4-a
hardware: CPU - e2-standard-4
stack: Python / FastAPI
trained on: router over digits 0-9
GPU expert
us-central1-a
hardware: GPU-T4 16GB
stack: JAX / XLA (CUDA)
trained on: digits 5-9
Digit
Steps
Routing
top-1 router
ensemble (weighted)
random pick
Generate