Machine Learning Compiler Engineer
Lightmatter, United States
Job Description
Lightmatter builds chips for artificial intelligence computing. Our architecture leverages unique properties of light to enable fast and efficient inference and training engines. If you're a collaborative engineer or scientist who has a passion for innovation, solving challenging technical problems and doing impactful work.work like building the world's first optical computers, consider joining the team at Lightmatter!
We are looking for talented compiler engineers to help us build the next generation of AI processors.
In this role, you will be responsible for developing highly optimized computational methods for deep learning. You'll be working with deep learning scientists, as well as digital, analog, and photonic designers, to implement the features needed to accelerate the next generation of machine learning algorithms.
Responsibilities
- Develop and implement compiler tools that translate deep learning packages (e.g. Tensorflow or PyTorch) directly to optimized native code.
- Analyze, craft, and implement compiler and linear algebra optimization techniques that exploit novel architectures and platforms being created at Lightmatter.
- Work with digital and analog engineers to define instruction sets that enable low-latency, high-throughput deep learning computation.
- Evaluate trade-offs and optimize between different software-hardware architectures in terms of performance and efficiency.
Requirements
- MS or higher in computer science, computer engineering, electrical engineering, or related field.
- Experience with writing passes on LLVM or other standard Internal Representations.
- Experience with back-end optimizations.
- Highly proficient in C/C++ with a strong software architecture background.
- Proficient in hardware-software integration with accelerated multi-threaded and concurrent programming.
- Understanding of the different computing architectures, e.g. CPU, GPU, FPGA.
- Understanding of linear algebra and GEMM.
- Experience with Machine Learning Frameworks and their IRs (e.g. XLA, MLIR, or Glow) is a strong plus.
Benefits
- Health Care Plan (Medical, Dental amp; Vision)
- Retirement Plan 401k
- Life Insurance
- Paid Time Off
- Family Leave
- Stock Option Plan