Software Engineer, Staff - AI Simulator Tools
March 16
🏢 In-office - Bay Area
Description
• The role requires you to be part of the team that helps productize the SW stack for our AI compute engine. • As part of the Software team, you will be responsible for the development, enhancement, and maintenance of the next-generation AI hardware simulation tools for hardware and for developing software kernels for the hardware. • You possess experience building functional simulators for new HW architectures. • You possess a very strong understanding of various hardware architectures and how to map algorithms to the architecture. • You understand how to map computational graphs generated by AI frameworks to the underlying architecture. • You have had past experience working across all aspects of the full stack tool chain and understand the nuances of what it takes to optimize and trade-off various aspects of hardware-software co-design. • You are able to build and scale software deliverables in a tight development window. • You will work with a team of compiler experts to build out the compiler infrastructure working closely with other software (ML, Systems) and hardware (mixed signal, DSP, CPU) experts in the company.
Requirements
• MS or PhD preferred in Computer Science, Electrical Engineering, Math, Physics or related degree. with 5+ years of Industry Experience. • Strong grasp of computer architecture, data structures, system software, and machine learning fundamentals. • Proficient in C/C++ and Python development in Linux environment and using standard development tools. • Experience implementing functional simulators in high level languages such as C/C++, Python. • Self-motivated team player with a strong sense of ownership and leadership. Desired: • Prior startup, small team or incubation experience. • Experience implementing algorithms for specialized hardware such as FPGAs, DSPs, GPUs, AI accelerators. • Experience with ML algorithms and frameworks such as PyTorch and/or TensorFlow • Experience with ML compilers and frameworks such as MLIR, LLVM, TVM, GLow. • Experience with a deep learning framework (such as PyTorch, Tensorflow) and ML models for CV, NLP, or Recommendation. • Work experience at a cloud provider or AI compute / sub-system company.
Benefits
• MS or PhD preferred in Computer Science, Electrical Engineering, Math, Physics or related degree. with 5+ years of Industry Experience. • Strong grasp of computer architecture, data structures, system software, and machine learning fundamentals. • Proficient in C/C++ and Python development in Linux environment and using standard development tools. • Experience implementing functional simulators in high level languages such as C/C++, Python. • Self-motivated team player with a strong sense of ownership and leadership.
Apply Now
