β Remote Jobs & AI Opportunities
A short application guide for this role.
About this opportunity
Join a leading AI lab's cutting-edge GenAI team and help build foundational AI models from the ground up. We're seeking MLOps and ML systems engineers with hands-on experience in large language model infrastructure across any of four areas: accelerator kernel programming, performance profiling and trace…
What you would work on
- Design challenging tasks across kernels, profiling, debugging and inference serving, including adversarial problems that expose where frontier models fail on accelerator code (for example Pallas kernels on TPU), and write…
- Guide research and engineering teams to close knowledge gaps and improve AI model performance on ML systems, compilers and training infrastructure topics.
- Evaluate ML systems tasks and solutions, and provide clear written technical feedback that stands up to expert review.
- Develop guidelines and detailed rubrics covering kernel optimization, profiler output interpretation, distributed systems reasoning, and serving throughput and latency trade-offs.
Check your fit
- 2+ years of hands-on professional experience in ML systems, ML infrastructure, model serving, or accelerator performance engineering. This is a hands-on systems role rather than an applied modelling or data…
- Practical experience in at least one of the following, with more than one a strong plus: writing or optimizing custom kernels (Pallas, Triton, CUDA, CUTLASS); performance profiling and trace analysis…
- Working production experience with JAX and/or PyTorch. Framework-level depth is a strong plus: custom operators, sharding and distributed training (pjit/shard_map, FSDP, Megatron, MaxText), or compiler and graph-level work (XLA/StableHLO, torch.compile,…
- Experience with modern accelerators such as TPU (v4 to v6e), H100, B200, Trainium or MI300, and the ability to reason about throughput, latency and memory trade-offs.
Prepare your application
- Tailor your CV to the relevant work and skills highlighted above.
- Prepare a brief example showing your relevant experience and the result you achieved.
- Confirm country eligibility, working hours, pay terms and any assessments on the application page.
Application
Read the complete listing and submit your application through the hiring platformβs application flow.
Continue to application βExplore more opportunities
AITraining.jobs
Explore AI roles across multiple hiring platforms, including roles listed for your country.
Explore AI roles βmicro1
Browse micro1βs current roles and find another opportunity that matches your experience.
Explore micro1 roles βTerac
Explore paid expert research and AI-related projects on Terac.
Explore Terac projects βSource: AITraining.jobs
