SIGN IN
Engineer, AI Framework Software (ML Accelerator Compiler Development) jobs in United States
cer-icon
Apply on Employer Site
company-logo

Black Sesame Technologies Inc · 15 hours ago

Engineer, AI Framework Software (ML Accelerator Compiler Development)

Black Sesame Technologies Inc is seeking an ML Accelerator Compiler Developer responsible for designing and optimizing compilers for Machine Learning accelerators. The role involves enhancing performance and efficiency of ML models through the development of tailored compiler toolchains and optimizations for specialized hardware architectures.
Artificial Intelligence (AI)AutomotiveMachine Learning
check
H1B Sponsor Likelynote
Hiring Manager
Jean Li, SPHR, SHRM-SCP
linkedin

Responsibilities

Develop and optimize compiler toolchains for ML accelerators, including front-end parsing, intermediate representation (IR) transformations, and backend code generation
Implement and enhance ML-specific optimizations such as operator fusion, memory layout transformations, quantization-aware compilation, and scheduling
Collaborate with hardware architects to co-design compiler optimizations aligned with accelerator capabilities
Work on ML frameworks (PyTorch, ONNX) to integrate compiler passes for efficient execution on target hardware
Improve performance through domain-specific optimizations, autotuning, and parallelization techniques
Debug and analyze performance bottlenecks across software and hardware stacks
Develop automated testing, benchmarking, and profiling tools for validating compiler optimizations

Qualification

Compiler developmentMachine Learning optimizationC++ programmingPython programmingParallel computingDeep learning frameworksPerformance profilingGraph optimizationsAssembly-level optimizationsMemory hierarchy optimizationsHeterogeneous computingAI model deploymentHigh-performance computing

Required

Strong proficiency in compiler development, including experience with LLVM, MLIR, TVM, or similar frameworks
Expertise in Machine Learning model execution, optimization, and deployment
Strong programming skills in C++, Python, and assembly-level optimizations
Knowledge of parallel computing, vectorization, and memory hierarchy optimizations
Familiarity with deep learning frameworks (TensorFlow, PyTorch, ONNX)
Strong analytical skills for performance profiling and debugging
Experience in graph optimizations, quantization, and code generation

Preferred

Knowledge of heterogeneous computing, DSPs, and low-level hardware programming
Familiarity with AI model deployment and inference optimization techniques
Background in high-performance computing (HPC)

Company

Black Sesame Technologies Inc

twittertwitter
company-logo
Founded in July 2016, Black Sesame Technologies is an AI digital imaging technology firm that creates solutions for real-world AI challenges.

H1B Sponsorship

Black Sesame Technologies Inc 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
2025 (19)
2024 (14)
2023 (25)
2022 (16)
2021 (20)
2020 (11)

Funding

Current Stage
Public Company
Total Funding
$345.69M
Key Investors
Summitview CapitalDongfeng Asset ManagementBoyuan Capital
2026-01-09Post Ipo Equity· $71.6M
2025-02-19Post Ipo Equity· $159.09M
2024-08-08IPO

Leadership Team

leader-logo
Wilson Liu
Co-Founder and COO
linkedin
Company data provided by crunchbase