Sr. Research Engineer, ML for AGI - Multimodal LLMs
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
Lead the development of state-of-the-art multimodal LLMs from pre-training to post-training workflows.
Scale model training on massive GPU and AWS Trainium clusters, optimizing distributed training and parallelism.
Refine low-level training stack components (CUDA kernels, collectives, network I/O) for high efficiency.
Build, extend, and evaluate using leading frameworks (NeMo, Megatron Core, PyTorch, Jax, vLLM, TRT).
Collaborate across teams to prototype novel techniques, assess feasibility, and deliver robust AI systems.
Operate in a fast-moving Agile environment, shaping architecture and best practices for reliable GenAI products.
Scale model training on massive GPU and AWS Trainium clusters, optimizing distributed training and parallelism.
Refine low-level training stack components (CUDA kernels, collectives, network I/O) for high efficiency.
Build, extend, and evaluate using leading frameworks (NeMo, Megatron Core, PyTorch, Jax, vLLM, TRT).
Collaborate across teams to prototype novel techniques, assess feasibility, and deliver robust AI systems.
Operate in a fast-moving Agile environment, shaping architecture and best practices for reliable GenAI products.
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.