Member of Technical Staff, Training Performance Engineer in London

Member of Technical Staff, Training Performance Engineer in London

London Full-Time 36000 - 60000 £ / year (est.) Remote
C

At a Glance

  • Tasks: Optimise performance of advanced language models and systems for AI training.
  • Company: Cohere is on a mission to scale intelligence for humanity through innovative AI solutions.
  • Benefits: Enjoy remote flexibility, weekly lunch stipends, full health benefits, and 6 weeks of vacation.
  • Other info: Diversity is celebrated; all backgrounds are encouraged to apply.
  • Why this job: Join a passionate team at the forefront of AI research and make a real impact.
  • Qualifications: Strong software engineering skills, proficiency in Python, and experience with GPU kernels required.

The predicted salary is between 36000 - 60000 £ per year.

Requirements

  • Extremely strong software engineering skills
  • Proficiency in Python and related ML frameworks such as JAX, Pytorch and XLA/MLIR
  • Experience writing kernels for GPUs using CUDA, triton, etc
  • Experience using large-scale distributed training strategies
  • Familiarity with autoregressive sequence models, such as Transformers
  • (Desirable) Bonus: paper at top-tier venues (such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP)
  • If some of the above doesn’t line up perfectly with your experience, we still encourage you to apply!

What the job involves

  • As a Performance Engineer in the Pre-Training team you will be responsible for optimizing the performance of our advanced language models and systems
  • Their primary focus is on improving key model training metrics, such as training throughput, ensuring high accelerator utilization
  • The team combines expertise in software engineering, machine learning, and low-level kernel design and development to design robust systems and enhance model performance
  • You will work on identifying and removing performance bottlenecks, develop cutting‑edge training and profiling tools to help Cohere's mission of providing efficient and reliable language understanding and generation capabilities and drive innovation in the field of natural language processing
  • Design and write high-performant and scalable software for training
  • Understand architectural modifications and design choices and their effects on training throughput and quality
  • Write low-level CUDA, triton kernels to squeeze every last bit of performance from our accelerators
  • Research, implement, and experiment with ideas on our supercompute and data infrastructure
  • Learn from and work with the best researchers in the field

#J-18808-Ljbffr

Member of Technical Staff, Training Performance Engineer in London employer: Cohere

Cohere is an exceptional employer that champions innovation and compliance in the rapidly evolving field of AI. With a strong commitment to employee growth, we offer a collaborative work culture where your expertise in EU regulations will directly influence our cutting-edge projects. Located in a vibrant tech hub, our team enjoys unique networking opportunities and access to industry leaders, making it an ideal place for professionals looking to make a meaningful impact.

C

Contact Details:

Cohere Recruitment Team

StudySmarter Expert Advice🤫

We think this is how you could land Member of Technical Staff, Training Performance Engineer in London

✨Join Local Tech Meetups

Get out there and mingle with fellow developers by joining local tech meetups. It’s a fantastic way to meet people who might be working at Cohere or know someone who does. Plus, you can pick up some trendy tech skills and trends while you're at it!

✨Contribute to Open Source Projects

Show off your coding chops by jumping into open-source projects. Not only does this give you practical experience, but it also gets you noticed in the dev community. You'll create a killer portfolio that speaks volumes about your skills to Cohere.

✨Tap into Online Developer Communities

Don’t underestimate the power of online developer communities like GitHub, Stack Overflow, and even Reddit. Participate in discussions, share your projects, and build your visibility. We can often find opportunities through these channels that can lead to a full-time gig at companies like Cohere.

✨Explore Job Boards Specifically for Tech Roles

Keep your eyes peeled on job boards that focus on tech roles. Sites like TechCareers or Stack Overflow Jobs can often have listings for companies like Cohere that might not show up on broader job sites. Make it a habit to check these regularly, and don’t hesitate to apply directly through our website!

We think you need these skills to ace Member of Technical Staff, Training Performance Engineer in London

Strong Software Engineering Skills
Proficiency in Python
Experience with ML Frameworks (JAX, PyTorch, XLA/MLIR)
CUDA Programming
Experience with Triton Kernels
Large-Scale Distributed Training Strategies
Understanding of Autoregressive Sequence Models (Transformers)

Some tips for your application 🫡

Show off your coding skills:When applying for a software engineering role, it's super important to showcase your coding skills. Make sure your CV includes your tech stack, any relevant programming languages you’re comfortable with, and examples of projects you've worked on. If you have a GitHub profile, link it up! We love to see code in action.

Tailor your portfolio:For a full-time role, we’d expect to see some solid examples of your work in your portfolio. Make sure to include at least two or three projects that highlight your problem-solving skills and your ability to work with different technologies. Focus on the projects that are most relevant to the position at Cohere.

Craft a killer cover letter:Your cover letter is your chance to stand out—make it personal! Explain why you want to work at Cohere and how your skills align with the role. Show us your passion for software development. We dig enthusiastic candidates who understand the value of collaboration and continuous learning!

Be clear and concise:When it comes to writing your CV and cover letter, clarity is key. Avoid jargon that could confuse us and stick to simple, direct language. Highlight your achievements with quantifiable results where possible, and keep everything easy to read. A well-organised application goes a long way!

How to prepare for a job interview at Cohere

✨Brush Up on Your Coding Skills

For a full-time software engineering role, it's crucial that we stay sharp with our coding abilities. Expect technical questions that might involve solving problems on the spot or discussing algorithms. Practise on platforms like LeetCode or HackerRank to get comfortable with the types of questions that often come up.

✨Know Your Tools and Frameworks

Make sure we’re well-acquainted with the tools and technologies listed in the job description. Familiarise ourselves with any specific frameworks or programming languages mentioned. If Cohere uses React or Node.js, for instance, be ready to discuss how we’ve used them in previous projects or coursework.

✨Showcase Your Projects

Bring along a portfolio that highlights our best work. This could be code samples, GitHub repositories, or any side projects we’ve built. Make sure we can talk through our thought process for each project, especially the challenges we faced and how we solved them—this shows our problem-solving skills in action.

✨Prepare for Behavioural Questions

While technical skills are key, full-time positions also require cultural fit. Be ready to discuss our previous experiences and how we handle teamwork, conflict, and deadlines. Brush up on the STAR method—Situation, Task, Action, Result—to clearly articulate our past experiences when discussing how we've contributed to a team.