SIGN IN
ML Compiler Software Engineer jobs in United States
info-icon
This job has closed.
company-logo

Symbolica AI · 4 months ago

ML Compiler Software Engineer

Symbolica AI is an AI research lab focused on applying category theory to create logical reasoning models in machines. They are seeking a Founding ML Compiler Engineer to lead the development of a compiler stack and GPU kernels for their custom dependently typed language, which aims to enable efficient AI architecture specifications.
Artificial Intelligence (AI)Software
check
H1B Sponsor Likelynote

Responsibilities

Translate high-level symbolic architecture specs (written in our custom dependently typed DSL) into efficient compute graphs and GPU-executable code
Build and optimize GPU kernels using CUDA and Rust, targeting training and inference of symbolic AI models
Design and implement compiler infrastructure (e.g. custom IRs, graph lowering, scheduling, memory planning) using MLIR, LLVM, or your own abstractions
Collaborate with mathematicians and researchers to co-design the system from first principles, ensuring semantic correctness throughout
Profile and debug across the stack — from type-level constructs to kernel performance — ensuring mathematical expressiveness meets real-world throughput

Qualification

RustCUDACompilersGPU kernelsDependent typesFunctional programmingCategory theoryType theory

Required

Strong experience with Rust or other performant system languages (e.g. C++, Haskell, Julia)
Expertise in compilers, intermediate representations, and building static analyses or program transformations
Familiarity with dependent types, symbolic computation, or strongly typed DSLs
Experience with CUDA, GPU kernels, and performance tuning at the memory/threading level

Preferred

Background in functional programming, category theory, or type theory

Benefits

Competitive salary and early-stage equity package.
A high-trust, execution-first culture with minimal bureaucracy.
Direct ownership of meaningful projects with real business impact.
A rare opportunity to sit at the interface between deep research and real-world productisation.

Company

Symbolica AI

twittertwittertwitter
company-logo
Symbolica AI operates as a research-focused company.

H1B Sponsorship

Symbolica AI has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2026 (1)

Funding

Current Stage
Early Stage
Total Funding
$33M
Key Investors
Khosla Ventures
2024-04-09Series A· $33M
2024-04-09Seed

Leadership Team

leader-logo
George Morgan
Founder and CEO
linkedin
Company data provided by crunchbase