MTS, Research Engineer

Fireworks AI
San Mateo, New York
Workplace: OnsiteFull timeUSD 250,000 - 400,000 annuallyFunction: Research & Scientific (R&D)Skills: ["Collaboration","Ownership","Experiment design","Problem-solving"]

Build and scale distributed training infrastructure while conducting open-ended research to improve model architectures, training objectives, and optimization techniques. Reproduce and extend state-of-the-art results from machine learning papers, then translate research ideas into robust, efficient, bug-free code. Collaborate with research scientists to unblock experiments with hardware-aware tooling, optimizing training loops, data pipelines, and communication across large GPU clusters.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Fireworks AI
Fireworks AI
4 weeks ago

MTS, Research Engineer

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 4 hours agoStatus: Live
Reposted: similar role first listed 1 week ago

Job Summary

Build and scale distributed training infrastructure while conducting open-ended research to improve model architectures, training objectives, and optimization techniques. Reproduce and extend state-of-the-art results from machine learning papers, then translate research ideas into robust, efficient, bug-free code. Collaborate with research scientists to unblock experiments with hardware-aware tooling, optimizing training loops, data pipelines, and communication across large GPU clusters.
Location: San Mateo, New York
Workplace: Onsite
Employment Type: Full time
Job Function: Research & Scientific (R&D)

Key Responsibilities

  • •Conduct open-ended research on model architectures, training objectives, and optimization techniques.
  • •Reproduce and extend state-of-the-art results from recent ML papers; identify bottlenecks and scale to larger datasets/models.
  • •Design, implement, and maintain high-performance distributed machine learning systems, optimizing training loops, data loaders, and GPU-cluster communication.
  • •Translate research ideas and mathematical concepts into robust, efficient, bug-free code.
  • •Collaborate with research scientists on tooling, code optimization, and hardware-aware co-design to unblock experiments.

Pay and Benefits

Salary: USD 250,000 - 400,000 annually
Equity and Bonus:Equity

Key Requirements

  • •Strong programming skills (Python, C++, or Rust) and a commitment to clean, maintainable code.
  • •Deep practical knowledge of machine learning frameworks (PyTorch, JAX, or TensorFlow).
  • •Experience with large distributed systems and parallel computing (CUDA, NCCL, MPI).
  • •A strong foundation in linear algebra, calculus, probability, and statistics.
  • •A proven track record of implementing complex deep learning algorithms from scratch.
Experience:Deep learning
Skills:CollaborationOwnershipExperiment designProblem-solving
Tech Stack:PythonC++RustPyTorchJAXTensorFlowCUDANCCLMPITritonCUDA/TritonSGLangVLLMLLMsGPUDistributed trainingData pipelines

Company Brief

Fireworks AI
Develops AI-driven tools to generate and optimize visual marketing content for brands and creators, automating production of short-form videos and multimedia assets for social platforms to improve engagement and scale creative workflows.
Industry: SaaS
Website