This job is closed
Applications are no longer available for this announcement. Explore current related opportunities below.
Job description
Xoriant Corporation in San Jose, CA is seeking an experienced Senior Staff Engineer to lead PyTorch accelerator integration and upstream contributions. This role focuses on upstreaming hardware support, architecture alignment, and collaboration with PyTorch maintainers.
You will design backend integration, enhance ATen and torch.compile workloads, and drive testing, benchmarks, and documentation across Python/C extensions and GPU accelerators.
#J-18808-Ljbffr
Worksite address
san jose, CA, 95199, US
Who can apply
Review the original listing for work authorization, qualifications and employer requirements.