AI Compiler职位-湾区自动驾驶芯片公司

职位-ML Accelerator Compiler Developer : NG 友好，需要有compiler经验，c++熟练

Job description:

As an ML Accelerator Compiler Developer, you will be responsible for designing and optimizing compilers for Machine Learning (ML) accelerators. You will work on enhancing performance, efficiency, and portability of ML models by developing compiler toolchains, optimizations, and code generation techniques tailored for specialized hardware architectures. 

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.

Qualifications:

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.

公司简介

• 已上市，芯片已量产
• 美国是核心研发团队，约 100 人，新加坡/中国亦有岗位

工作地点

• San Jose - Fully onsite

对自动驾驶行业和 AI 芯片感兴趣的朋友，欢迎将简历发送至 HR 邮箱：ann.gao@bst.ai

所以公司叫BST吗

已发，dav***ong@gmail.com，感谢楼主！