Research Engineer, Artificial General Intelligence
Is this job for you?
Build my CV Build your CV and see how well you match this role — and every other one.
The role
We are seeking a hands-on Research Engineer to advance multimodal large-language foundational models (LLMs) at scale.
You will design and implement novel algorithms and modeling techniques for pre-training and post-training workflows.
Scale model training on hyper-large GPU clusters and AWS Trainium, while optimizing distributed training and parallelism.
Refine low-level training stack components, including CUDA kernels, communication collectives, and network I/O, to maximize efficiency.
Utilize and extend industry-leading frameworks (NeMo, Megatron Core, PyTorch, Jax, vLLM, TRT) and evaluate technical feasibility.
Deliver results independently in a self-organizing Agile environment while adapting to new scientific advances and collaborating with teammates.
You will design and implement novel algorithms and modeling techniques for pre-training and post-training workflows.
Scale model training on hyper-large GPU clusters and AWS Trainium, while optimizing distributed training and parallelism.
Refine low-level training stack components, including CUDA kernels, communication collectives, and network I/O, to maximize efficiency.
Utilize and extend industry-leading frameworks (NeMo, Megatron Core, PyTorch, Jax, vLLM, TRT) and evaluate technical feasibility.
Deliver results independently in a self-organizing Agile environment while adapting to new scientific advances and collaborating with teammates.
See the full job post
Responsibilities, requirements, skills and benefits — create your free account.
or
Already have an account?
Log inSimilar openings
Other roles that could suit you.
? Fellow, AI Performance Software Engineer ? Applied Machine Learning Engineer, Circuit Design - New College Grad 2026 ? Applied Machine Learning Engineer - VLSI Design ? Machine Learning Engineer II, Sponsored Products and Brands ? Software Engineer- AI/ML, AWS Neuron Distributed Training ? Applied Scientist, LLM Code Agents, Kiro Science