Job Description
the company will be prioritizing applicants who have a current right to work in Singapore, and do not require the company's sponsorship of a visa.
Minimum qualifications- Bachelor's degree or equivalent practical experience.
- 10 years of experience in developing and deploying models using deep learning frameworks (e.g., TensorFlow, PyTorch, or JAX) specifically on TPU hardware.
- Experience in networking principles, including concepts like collective communication, inter-chip interconnects, and their impact on distributed AI training.
- Experience with lower-level performance tools and techniques (e.g., custom kernel development, XLA compiler familiarity) relevant to optimizing code for the company's TPU chips.
- Experience with leveraging AI hardware and software stacks and platforms to bring up and deploy AI compute clusters.
- Knowledge of AI accelerator hardware (e.g., specific GPU ge...
Apply for This Position
Ready to take the next step? Click the button below to submit your application.
Submit Application