R

Reflection AI

Builds AI models and deployment software

Member of Technical Staff - Pre-Training

AI Research +1Confirmed live in the last 24 hours
Full-timeIn-person

About the Job

Requirements

  • A graduate degree (MS or PhD) in Computer Science, Machine Learning, or a related discipline.
  • Solid software engineering capabilities and experience building large-scale systems.
  • Experience with large-scale ETL workflows and training data preparation.
  • Deep understanding of large-scale machine learning, especially language models, distributed training, and scaling.
  • Proficiency in Python and deep learning frameworks; PyTorch is preferred.
  • Ability to navigate trade-offs between research objectives and practical engineering realities.

Responsibilities

  • Research and build solutions across algorithms, scaling laws, data processing, optimizers, and model architecture.
  • Design and run scientific experiments to advance understanding of scaling large language models and data efficiency.
  • Implement state-of-the-art methods from the deep learning literature.
  • Lead small research projects independently while collaborating on larger initiatives.
  • Optimize the training infrastructure for efficient scaling.
  • Contribute across the entire stack, from low-level optimizations to high-level model design.

Desired Qualifications

  • Proficiency in PyTorch.

About the Company

Reflection AI

Builds AI models and deployment software

Reflection AI develops language models for software developers, companies and governments that want control over how AI is deployed and adapted. Its research concentrates on coding, reasoning and AI agents that work through tasks with software tools. The lab builds the surrounding software and infrastructure as well as models, letting customers run AI through its hosted platform or in their own computing environments. Its commercial work pairs model access with engineers who design applications and support customized enterprise deployments from development through operation. Reflection AI was founded in 2024 and is headquartered in New York City.

Company size
201-500
Company stage
Late Stage VC
Total funding
$4.6B
Headquarters
New York City, New York
Founded
2024

Recently Posted Jobs

Explore more opportunities at Reflection AI

See more jobs