High Performance Computing (Triton + MPI) Engineer

Key Skills

Aartificial intelligence, High performance computing- HPC, Large Language Models - LLMs, Machine learning techniques

Job Description

Lamini AI is at the forefront of bringing LLMs to production. We are on a mission to help every company unlock the power of generative AI, by putting their own data to work. Our team is made up of highly experienced ML engineers and tech industry veterans and we’re backed by leading computing and technology companies.

HPC Engineers in our team are responsible for one or more of: developing and optimizing high performance collective and kernel libraries for running LLMs on AMD GPUs, using technologies including MPI, ROCe, UCX, collectives, OpenAI Triton, etc.

At Lamini AI, we are committed to providing an environment of mutual respect where equal employment opportunities are available to all applicants without regard to race, color, religion, sex, pregnancy (including childbirth, lactation and related medical conditions), national origin, age, physical and mental disability, marital status, sexual orientation, gender identity, gender expression, genetic information (including characteristics and testing), military and veteran status, and any other characteristic protected by applicable law. Lamini AI believes that diversity and inclusion among our employees is critical to our success as a company, and we seek to recruit, develop and retain the most talented people from a diverse candidate pool. Selection for employment is decided on the basis of qualifications, merit, and business need.