AI Performance Engineer (Cloud AI Engineering), Sr | Staff | Sr. Staff - Cork, Ireland

This role involves the following activities:

• Convert,optimizeand deploy models for efficient inference usingPyTorch, ONNX.

• Work at the forefront ofGenAIby understanding advanced algorithms (e.g.attention mechanisms,MoEs) andnumericstoidentifynew optimization opportunities.

• Performance analysis and optimization of LLM, VLM, and diffusion models for inference.  Scale performance for throughput and latency constraints.

• Mapping the next generation AI workloads on top of current and future hardware designs. 

• Work closely with customers to drive solutions by collaborating with internal compiler,firmwareand platform teams.

• Analyze complex performance or stability issues to work towards final root cause of underlying problems.

• Create engineering solutions to deliver continuous insights into performance of AI workloads guiding the improvements over time.

• Design and implement high-level kernels,e.g.in Triton, with a focus on generating efficient, low-level code.

Candidates for this position willdemonstratethe following:

• Hands-on experience in building andoptimizinglanguage models, notably inPyTorch, ONNX, preferably in production-grade environments.

• Deep understanding of transformer architectures, attentionmechanismsand performance trade-offs.

• Experience in workload mapping strategies exhibiting sharding or various parallelisms.

• Strong Python programming skills.

• Proactive learning about the latest inference optimization techniques.

• Understanding of computer architecture, ML accelerators, in-memoryprocessinganddistributedsystems.

• Strong communication, problem-solvingskillsand ability to learn and work effectively in a fast-pacedandcollaborative environment.

• MS in Computer Science, Machine Learning, Computer Engineering or Electrical Engineering.

Bonus Skills:

• Background in neural network operators and mathematical operations, including linear algebra and math libraries.

• Understanding of machine learning compilers.

• Experience in converging accuracy and its evaluation methods.

• Knowledge of torch.compileortorchDynamo.

• PhD in Computer Science, Computer EngineeringorMachine Learning

Where you will be working

Cork has a proud reputation as Ireland's second largest economic engine and is now one of the Top 20 location choices in Europe with 39,000 people being employed by over 170 overseas companies.
There's a growing diversity in the region with people from many nationalities relocating to Cork, relishing the opportunity to work and live in a location that offers an excellent quality of life.
A gateway to Europe, Cork airport provides access to almost 50 international destinations including transatlantic air routes.

Equal Opportunities
We are an Equal Opportunity employer; all qualified applicants will receive consideration for employment without regard to race, colour, religion, sexual orientation, gender identity, national origin, disability, veteran status, or any protected classification.

What's on Offer

Apart from working in an open, relaxed and collaborative space, you will enjoy:

• Salary, stock and performance related bonus

• Maternity/Paternity Leave

• Employee stock purchase scheme

• Matching pension scheme

• Education Assistance

• Relocation and immigration support (if needed)

• Life, Medical, Income and Travel Insurance

• Subsidised memberships for physical and mental well-being

• Bicycle purchase scheme

• Employee run clubs, including, running, football, chess, badminton + many more