Software Engineer, AI accelerators
- PyTorch/ATen backend paths for a photonic AI accelerator: custom kernels, graph execution, and framework integration.
- Runtime dispatch for graph partitioning, fast allocation, peer transfers, launch serialization, and concurrent queues.
- Host code tied to firmware flows where launches become primitive device graphs for mono cores and sync points.
- Hardware-adjacent work: custom ISA emission, driver interfaces, device setup, simulation, and validation.