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.