gem
AI Runtime Engineer
Modular AI
Edinburgh, GBR
Full-time
Entry · 2+ years experience
Hybrid
GBP 82800-123600 yearly / year
Discovered Yesterday
C++MLIRLLVMMojo
Free
Job Fit Check
Base Career helps you apply smarter for this job.
?%
Ready to ScanKey skills for this role
C++MLIRLLVM
Key Skills for This Role
C++MLIRLLVMMojo
Full Job Posting
What you will do:
- Design and develop runtime and cross-stack optimizations to improve CPU, GPU, and accelerator efficiency, addressing issues such as CPU overhead, caching, and data locality across multiple devices.
- Work with vendor-specific networking libraries to unlock high performance data transfer for multiple topologies.
- Collaborate with the compiler, kernels, serving, and models teams to design core technologies that achieve state-of-the-art end-to-end performance on various CPU and GPU hardware.
- Collaborate with the customer success team and engage with customers to understand their performance requirements and use cases.
- Collaborate with tooling and infrastructure teams to design systems for automated performance analysis and benchmarking.
What you bring to the table:
2+ years of experience working on high-performance computing systems.
Experience in C++ programming and complex software systems.
Experience with CPU or GPU runtime optimizations and performance analysis on CPUs, GPUs, or AI accelerators.
Proficiency with one or more profiling tools (CPU or GPU).
Creativity and curiosity for solving complex problems, a team-oriented attitude that enables you to work well with others, and alignment with our culture.
Helpful, but not required:
Experience with ML graph optimizations, parallel / distributed programming, heterogeneous ML computation, and/or code generation.
Exposure to MLIR, LLVM, and/or the Mojo programming language.
Advanced degree in Computer Science or a related area is a plus.
Apply for this job in 1 click
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer