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.
Track your applications on mobile The free Whileresume app, on iPhone and Android.
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 inYou might also like these jobs
No closely matching jobs yet — here are the most recent ones.