Enterprise GenAI ML Systems Research Engineer

United States Digital Space LLC; New York, San Francisco (NY, CA); On-site; USD 265,000 - 331,000; Full time

Job summary

United States Digital Space LLC in New York seeks an experienced ML Research Engineer to join the Enterprise ML Research Lab. You will build, profile and optimize our training and inference framework and post-train state-of-the-art models for enterprise engagements.

You will collaborate with ML teams to accelerate research, develop next-gen agent training algorithms for multi-agent rollouts, and advance architectures for large-scale GPU clusters and multi-node LLM workloads.

Qualifications

• At least 1–3 years of LLM training in a production environment.
• Experience with post-training methods like RLHF/RLVR and related algorithms.
• Ability to operate the architecture of a modern GPU cluster.
• Experience with multi-node LLM training and inference.

Responsibilities

• Build, profile and optimize our training and inference framework.
• Post-train state of the art models, developed both internally and from the community.
• Collaborate with ML teams to accelerate their research and development.
• Create a next-gen agent training algorithm for multi-agent/multi-tool rollouts.

Skills

LLM training

System optimization

Multi-node training

GPU cluster architecture

RLHF/RLVR

PPO/GRPO

Cross-functional collaboration

Strong communication

Education

PhD or Masters in Computer Science or related field

Tools

CUDA

PyTorch

Transformers

Flash attention

Job description

United States Digital Space LLC in New York seeks an experienced ML Research Engineer to join the Enterprise ML Research Lab. You will build, profile and optimize our training and inference framework and post-train state-of-the-art models for enterprise engagements.

You will collaborate with ML teams to accelerate research, develop next-gen agent training algorithms for multi-agent rollouts, and advance architectures for large-scale GPU clusters and multi-node LLM workloads.