Software Dev Engineer, Machine Learning Compilers
Core
Designing and implementing compiler infrastructure and optimization passes to accelerate deep learning networks on custom edge AI silicon and heterogeneous device backends.
Role type
Machine Learning Compiler Engineer
Builds
Compiler infrastructure, code generation backends, and optimization tooling for custom neural accelerators
Domain
Edge AI, Custom Silicon, Compiler Technology
Deliverable
production ML models
Required skills
Compiler design and development, graph optimization and partitioning, backend code generation for heterogeneous platforms, system-level performance profiling, LLM deployment on AI acceleration hardware (GPUs, Neuron, TPU), C++ programming, object-oriented design
Preferred skills
Embedded development, building compilers for application-specific accelerators, full software development lifecycle management, experience with novel hardware schedulers
Technologies
C++, Java, C#, GPUs, Neuron, TPU
Responsibilities
Design and develop software stack for deep learning accelerators, develop compiler passes for graph ingestion and optimization, profile and optimize system-level performance, participate in design reviews and API development, collaborate with hardware and science teams to onboard user experiences, mentor junior engineers
Seniority
Mid-Senior, hands-on IC
