Job Description
As a Member of Technical Staff, you will:
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.
You may be a good fit if you have:
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.
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.
You may be a good fit if you have:
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.
💡 Quick Summary
Seeking a career-building opportunity? The Member of Technical Staff, Training Performance Engineer position is now open for candidates interested in the IT Engineer & Developer Jobs sector. This role in London offers a professional environment and growth potential.
Requirement Snapshot: Candidates should possess basic communication skills, a proactive attitude, and the ability to work in a team. Experience in IT Engineer & Developer Jobs is a plus.