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
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
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
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 StageTotal Funding
$33MKey Investors
Khosla Ventures
2024-04-09Series A· $33M
2024-04-09Seed
Recent News
General Catalyst
2026-02-11
2026-01-16
Company data provided by crunchbase