Senior Ai Frameworks Engineer

Invidia

Base: $152,000 - $287,500 usd depending on level; ...
Python and c++ proficiency
Python extension and ffi design
Library or framework development experience
This role focuses on building the next frontier of the CUTLASS ecosystem by bringing high-performance abstractions directly into the Python environment

Job Summary

  • This role focuses on building the next frontier of the CUTLASS ecosystem by bringing high-performance abstractions directly into the Python environment.
  • The engineer will design APIs that prioritize user productivity while developing robust compilation infrastructure to lower Pythonic descriptions into GPU machine code.
  • Candidates will build production-grade delivery infrastructure including package distribution, documentation, and testing for the open-source community.

Matching Summary

This role focuses on building the next frontier of the CUTLASS ecosystem by bringing high-performance abstractions directly into the Python environment.

Salary

Base: $152,000 - $287,500 USD depending on level; Equity: Eligible; Benefits: Eligible

Skills & Requirements

Must-have

  • Python and C++ proficiency
  • Python extension and FFI design
  • Library or framework development experience
  • AST transformations and JIT execution
  • GPU programming and CUDA knowledge

Nice-to-have

  • Open-source library maintenance
  • Compiler foundations and IR understanding
  • Experience with LLVM or MLIR
  • Debugging tool creation
  • Passion for developer productivity

Key Requirements

  • MS or PhD in Computer Science or related field
  • At least 3+ years of relevant experience
  • Deep understanding of Python delivery stack

Work Rights

Not specified

Tailored Resume

Cover Letter