Design and develop a multi-target runtime, optimize kernels, analyze compiler outputs, and improve architecture based on ML engineers' needs.
At Lemurian Labs, we’re on a mission to bring the power of AI to everyone—without leaving a massive environmental footprint. We care deeply about the impact AI has on our society and planet, and we’re building a rock-solid foundation for its future, ensuring AI grows sustainably and responsibly. Because let’s face it, what good is innovation if it doesn’t help the world?
We are building a high-performance, portable compiler that lets developers “build once, deploy anywhere.” Yes, anywhere. We’re talking about seamless cross-platform compatibility, so you can train your models in the cloud, deploy them to the edge, and everything in between—all while optimizing for resource efficiency and scalability.
If the idea of sustainably scaling AI motivates you and you’re excited about making AI development both powerful and accessible, then we’d love to have you. Join us at Lemurian Labs, where you can have fun building the future—without leaving a mess behind.
Key Duties
- Design, develop, maintain and improve our multi-target runtime
- Use the latest techniques in parallelization and partitioning to automate generation and exploit highly optimized kernels
- Rapid prototyping and data driven exploration of new ideas
- Benchmark and analyze the outputs produced by our optimizing compiler on target hardware
- Work closely with our product team to understand the evolving needs of ML engineers and drive improvements in architecture
- Build tools to collect and analyze performance bottlenecks
Essential Skills and Experience
- A deep understanding of asynchronous, concurrent programming.
- 4+ years of experience with C/C++ (C++14 or newer).
- An understanding of HW architecture (vector vs scalar registers and instructions, memory hierarchies).
- Knowledge of operating system kernel development or hypervisor development.
Preferred Skills and Experience
- Experience developing or maintaining libraries like CUDA or ROCm.
- Experience with GPU programming.
- Experience with high performance computing (HPC).
- Masters or PhD degree in computer science, or equivalent practical experience.
- Knowledge of DL frameworks such as PyTorch, JAX or Triton.
- Experience with programming large compute clusters.
Salary depends on experience and geographical location.
This salary range may be inclusive of several career levels and will be narrowed during the interview process based on a number of factors, such as candidate’s experience, knowledge, skills and abilities, as well as internal equity among our team.
Additional benefits for this role may include: equity, company bonus opportunities; medical, dental, and vision benefits; retirement savings plan; and supplemental wellness benefits.
Lemurian Labs ensures equal employment opportunity without discrimination or harassment based on race, color, religion, sex (including pregnancy, childbirth, or related medical conditions), sexual orientation, gender identity or expression, age, disability, national origin, marital or domestic/civil partnership status, genetic information, citizenship status, veteran status, or any other characteristic protected by law.
EOE
Top Skills
C/C++
Cuda
Gpu
High Performance Computing
Jax
PyTorch
Rocm
Triton
Similar Jobs
Artificial Intelligence
Develop high-performance code for the distributed system, optimizing data structures and algorithms for Cerebras' architecture in AI applications.
Top Skills:
AssemblyC/C++Python
Artificial Intelligence
Lead the Inference ML team in developing tools and APIs for large-scale ML applications, enhancing performance and usability, while collaborating across engineering teams.
Top Skills:
C++JaxPythonPyTorchTensorFlow
Artificial Intelligence
Design and develop high-performance distributed software for scalable AI training systems, focusing on data pipelines and system efficiency.
Top Skills:
C/C++PythonPyTorch
What you need to know about the Calgary Tech Scene
Employees can spend up to one-third of their life at work, so choosing the right company is crucial, not just for the job itself but for the company culture as well. While startups often offer dynamic culture and growth opportunities, large corporations provide benefits like career development and networking, especially appealing to recent graduates. Fortunately, Calgary stands out as a hub for both, recognized as one of Startup Genome's Top 100 Emerging Ecosystems, while also playing host to a number of multinational enterprises. In Calgary, job seekers can find a wide range of opportunities.
