Machine Learning Compiler Engineer

Job Overview

Join Gensyn as a Machine Learning Compiler Engineer and be part of a team that is revolutionizing the way machine learning compute is accessed globally. This role is perfect for someone who is passionate about compiler engineering and has a strong background in deep learning and machine learning algorithms.

Responsibilities

Lower Deep Learning Graphs: Work with common frameworks like PyTorch, TensorFlow, and Keras to translate them into an intermediate representation (IR) for training, ensuring reproducibility.
Algorithm Development: Create novel algorithms for transforming intermediate representations of compute graphs between different operator representations.
Compiler Ownership: Take charge of two key areas:
- Front-end: Manage the integration of common Deep Learning Frameworks with Gensyn's IR, and write transformation passes in ONNX.
- Middle-end: Develop compiler passes for training-based compute graphs, integrate reproducible Deep Learning kernels, and debug compilation passes.
- Back-end: Lower IR from middle-end to GPU target machine code.

Minimum Requirements

Compiler Knowledge: Understanding of traditional compilers like LLVM or GCC and graph traversals.
Software Engineering Skills: Experience in shipping production code.
Parallel Programming: Knowledge of GPU programming.
Rust Proficiency: Willingness to learn and work with Rust.
Communication Skills: Excellent verbal and written communication.
Research Environment: Comfortable working with high autonomy in an applied research setting.

Nice to Haves

Architecture Understanding: Knowledge of computer architectures specialized for training neural network graphs.
Rust Experience: Systems-level programming experience in Rust.
Open-source Contributions: Experience contributing to compiler stacks.
Deep Learning Frameworks: Familiarity with frameworks like TVM, MLIR, or TensorComprehensions.

Benefits

Competitive Salary and Equity: Enjoy a competitive salary with a share of equity and token pool.
Remote Work: Fully remote position, hiring between the West Coast (PT) and Central Europe (CET) time zones.
Relocation Assistance: Available for those wishing to relocate within the specified time zones.
Company Retreats: Four all-expenses-paid company retreats per year.
Comprehensive Insurance: Private health, vision, and dental insurance, including coverage for dependents (US only).

About Gensyn

Gensyn is at the forefront of creating a global machine learning supercluster, making computational power as accessible as electricity. We operate with a high degree of autonomy, focusing on efficiency and rejecting mediocrity. Our small, agile teams are designed to rival much larger traditional teams, ensuring that we remain at the cutting edge of AI development.

If you are a highly motivated individual with a passion for machine learning and compiler engineering, we encourage you to apply and join us in shaping the future of AI.

Benefits
Extracted with AI

Dental insurance
Private health insurance
Vision insurance
Paid sick leave
Relocation assistance
Company retreats
Remote work

Similar jobs

Last update: 23 minutes ago

dataroots

Expert Machine Learning Engineer

Join Dataroots as an Expert Machine Learning Engineer to design and deliver AI-powered solutions, focusing on machine learning models.

7 days ago

Senior Backend Engineer C++

Join DeepL as a Senior Backend Engineer C++ to design and maintain scalable backend services using C++ and AI technologies.

7 days ago

Senior Backend Engineer - Java, Rust, Go

Join Together AI as a Senior Backend Engineer in Amsterdam. Work with Java, Rust, and Go to build scalable backend systems.

11 days ago

Senior Product Engineer [Rust & Typescript]

Join Attio as a Senior Product Engineer working with Rust & TypeScript to build innovative CRM features. Remote work available.

10 days ago

AI Engineer

Join BCG X as an AI Engineer in Milan, Italy. Develop AI solutions, partner with clients, and drive innovation in a dynamic environment.

7 days ago

Staff Software Engineer

Join Aiven as a Staff Software Engineer to develop cloud operations platforms using open-source technologies. Hybrid work in Berlin.

7 days ago

Senior Full Stack Engineer - Climate Tech - Rust & TypeScript

Join Climatiq as a Senior Full Stack Engineer to develop climate tech solutions using Rust and TypeScript. Remote work available.

6 days ago

Senior Software Engineer - .NET and Go

Join Sysdig as a Senior Software Engineer in Berlin, focusing on .NET and Go for cloud security solutions.

12 days ago

Senior Software Engineer - Backend Development

Join Sysdig as a Senior Software Engineer to develop scalable backend services using Go, RESTful APIs, and microservices in a hybrid work environment.

11 days ago

Full-Stack Engineer with Angular and React.js

Join Labelbox as a Full-Stack Engineer to develop scalable systems using Angular, React.js, and GraphQL. Work remotely in a dynamic AI-driven environment.

10 days ago