Bootstrap

Gensyn

Compiler Engineer - Distributed Machine Learning

๐Ÿ“ŒUnited States, United States ๐Ÿ‡บ๐Ÿ‡ธ

โฑ๏ธŽ full-time

๐Ÿง™โ€โ™‚๏ธ mid-level

remote

Machine intelligence will soon take over humanityโ€™s role in knowledge-keeping and creation. What started in the mid-1990s as the gradual off-loading of knowledge and decision making to search engines will be rapidly replaced by vast neural networks - with all knowledge compressed into their artificial neurons. Unlike organic life, machine intelligence, built within silicon, needs protocols to coordinate and grow. And, like nature, these protocols should be open, permissionless, and neutral. Starting with compute hardware, the Gensyn protocol networks together the core resources required for machine intelligence to flourish alongside human intelligence.


The Role

  • Lowering deep learning graphs - from common frameworks (PyTorch, Tensorflow, Keras, etc) down to an IR for training and inference - with particular focus on ensuring reproducibility


Responsibilities

  • Writing novel algorithms - for transforming intermediate representations of compute graphs between different operator representations
  • Owning two of the following compiler areas:
  • Front-end - deal with the handshaking of common Deep Learning Frameworks with Gensyn's IR for internal IR usage. Write Transformation passes in ONNX to alter IR for middle-end consumption
  • Middle-end - write compiler passes for training-based compute graphs, integrate reproducible Deep Learning kernels into the code generation stage, and debug compilation passes and transformations as you go
  • Back-end : lower IR from middle-end to GPU target machine code


Competencies

Must have

  • Compiler knowledge - base-level understanding of a traditional compiler (LLVM, GCC) and graph traversals required for writing code for such a compiler
  • Solid software engineering skills - practicing software engineer, having significantly contributed to/shipped production code
  • Understanding of parallel programming - specifically as it pertains to GPUs
  • Ability to operate on:
  • High-Level IR/Clang/LLVM up to middle-end optimisation; and/or
  • Low Level IR/LLVM targets/target-specific optimisations - particularly GPU specific optimisations
  • Highly self-motivated with excellent verbal and written communication skills
  • Comfortable working in an applied research environment with extremely high autonomy


Preferred

  • Architecture understanding - full understanding of a computer architecture specialised for training NN graphs (Intel Xeon CPU, GPUs, TPUs, custom accelerators)
  • Compilation understanding - strong understanding of compilation in regards to one or more High-Performance Computer architectures (CPU, GPU, custom accelerator, or a heterogenous system of all such components)
  • Proven technical foundation - in CPU and GPU architectures, numeric libraries, and modular software design
  • Deep Learning understanding - both in terms of recent architecture trends + fundamentals of how training works, and experience with machine learning frameworks and their internals (e.g. PyTorch, TensorFlow, scikit-learn, etc.)
  • Exposure to a Deep Learning Compiler frameworks - e.g. TVM, MLIR, TensorComprehensions, Triton, JAX
  • Kernel Experience - Experience writing and optimizing highly-performant GPU kernels


Nice to have

  • Open-source contributions to existing compilers/frameworks with a strong preference for ML compilers/frameworks.


Compensation / Benefits

  • Competitive salary + share of equity and token pool
  • Fully remote work - we hire between the West Coast (PT) and Central Europe (CET) time zones
  • Relocation Assistance - available for those that would like to relocate after being hired (anywhere from PST through CET time zones)
  • 3-4x all expenses paid company retreats around the world, per year
  • Whatever equipment you need
  • Paid sick leave
  • Private health, vision, and dental insurance - including spouse/dependents [๐Ÿ‡บ๐Ÿ‡ธ only]


Our Principles

Autonomy & Independence

  • Donโ€™t ask for permission - we have a constraint culture , not a permission culture.
  • Claim ownership of any work stream and set its goals/deadlines, rather than waiting to be assigned work or relying on job specs.
  • Push & pull context on your work rather than waiting for information from others and assuming people know what youโ€™re doing.
  • Communicate to be understood rather than pushing out information and expecting others to work to understand it.
  • Stay a small team - misalignment and politics scale super-linearly with team size. Small protocol teams rival much larger traditional teams.


Rejection of mediocrity & high performance

  • Give direct feedback to everyone immediately - rather than avoiding unpopularity , expecting things to improve naturally, or trading short-term pain for extreme long-term pain .
  • Embrace an extreme learning rate - rather than assuming limits to your ability / knowledge.
  • Donโ€™t quit - push to the final outcome, despite any barriers.
  • Be anti-fragile - balance short-term risk for long-term outcomes.
  • Reject waste - guard the companyโ€™s time, rather than wasting it in meetings without clear purpose/focus, or bikeshedding .
  • Build thinly - build and design thinly .

Other similar jobs

Compiler Developer

@ IBM, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

Senior Compiler Engineer - Backend

@ NVIDIA, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

Senior Compiler Engineer - Backend

@ NVIDIA, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

Senior Compiler Engineer - Backend

@ NVIDIA, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

Senior Compiler Engineer - Backend

@ NVIDIA, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

Staff GPU Compiler Engineer

@ Qualcomm, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

Software Dev Engineer - Compiler, Annapurna Labs

@ Amazon, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

Staff GPU Compiler Engineer

@ Qualcomm, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

Sr ML Compiler Engineer , Annapurna Labs

@ Amazon Web Services (AWS), ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

GPU compiler Engineer

@ Qualcomm, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

  • Employment

    โฑ๏ธŽ full-time

  • Experience

    ๐Ÿง™โ€โ™‚๏ธ mid-level

  • Working model

    remote

  • Skills
  • Industry
  • Find similar jobs

    Compiler Developer

    @ IBM, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

    Senior Compiler Engineer - Backend

    @ NVIDIA, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

    Senior Compiler Engineer - Backend

    @ NVIDIA, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

    Senior Compiler Engineer - Backend

    @ NVIDIA, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

    Senior Compiler Engineer - Backend

    @ NVIDIA, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

    Staff GPU Compiler Engineer

    @ Qualcomm, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

    Software Dev Engineer - Compiler, Annapurna Labs

    @ Amazon, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

    Staff GPU Compiler Engineer

    @ Qualcomm, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

    Sr ML Compiler Engineer , Annapurna Labs

    @ Amazon Web Services (AWS), ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

    GPU compiler Engineer

    @ Qualcomm, ๐Ÿ“United States ๐Ÿ‡บ๐Ÿ‡ธ

Remote Work
Post time
Level
Employment
Industry
Apply Now โ†—