Machine Learning Research Engineer

The AI age is upon us and high performance computing is the underlying platform powering everything from Large Language Models (LLM) to Image synthesis from text. However, with the demise of Moore’s law and Dennard scaling we are at an inflection point. At Lightmatter, we are leading the transition of computing from traditional electronic transistors to photonic technologies which can operate at mind blowing efficiency and throughput.

In this role, you will support all the activities of the ML team as it guides the development of a new class of computing infrastructure. This includes fine tuning LLMs, enablement of new models on custom architectures, evaluating the performance of models at scale, developing abstract models of the hardware for evaluating accuracy and throughput and help co-design novel hardware in a new paradigm of computing. Working in a small team of highly talented engineers, you will be able to move fast and develop well architected solutions.

If you are passionate about advanced AI technology and would like to develop scalable algorithms, hardware and ML techniques, join us!

Responsibilities

• Develop software for evaluating accuracy and throughput of models on a custom hardware.
• Develop performance models for a novel class of hardware.
• Develop scalable algorithms for large scale inference and training
• Train LLMs and other models on a large cluster of GPUs.
• Support ML researchers as they explore accuracy on a wide variety of models on custom hardware.
• Publish and present new research at premier ML/CS conferences.