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

Black Sesame Technologies Inc · 11 hours ago

Engineer, AI Framework Software (ML Accelerator Compiler Developer)

Black Sesame Technologies Inc is focused on developing advanced solutions for Machine Learning accelerators. The role involves designing and optimizing compilers specifically for ML accelerators, enhancing performance and efficiency through various compiler toolchains and optimizations.
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 programmingLLVM frameworkMLIR frameworkTVM frameworkDeep learning frameworksPerformance profilingAssembly-level optimizationsParallel computingVectorizationMemory hierarchy optimizationsGraph optimizationsQuantizationCode generationHeterogeneous computingDSPs programmingAI model deploymentInference optimizationHigh-performance computing

Required

Master's degree in Computer Science or a related field
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