MTS, Research Engineer

4 hours, 25 minutes ago
Full-time
Senior
DevOps and Infrastructure
Fireworks AI

Fireworks AI

Fireworks AI is a company that provides a generative AI platform as a service. They use state-of-the-art LLMs and image models to deliver fast and efficient AI solutions. Partnering with leading AI researchers, Fireworks offers models for various use c...

Internet Software & Services
1-10
Founded 2022

Description

  • Conduct open-ended research on model architectures, training objectives, and optimization techniques.
  • Formulate hypotheses, design experiments, and iterate based on empirical results.
  • Reproduce and extend state-of-the-art machine learning results from recent papers.
  • Identify bottlenecks and scale research methods to larger datasets and models.
  • Design, implement, and maintain high-performance distributed machine learning training systems.
  • Optimize training loops, data loaders, and communication overhead across large GPU clusters.
  • Translate research ideas and mathematical concepts into robust, efficient code.
  • Collaborate with Research Scientists to unblock experiments through tooling, code optimization, and hardware-aware experiment design.

Requirements

  • Strong programming skills in Python, C++, or Rust.
  • Commitment to writing clean, maintainable code.
  • Deep practical knowledge of machine learning frameworks such as PyTorch, JAX, or TensorFlow.
  • Experience with large distributed systems and parallel computing, including CUDA, NCCL, or MPI.
  • Strong foundation in linear algebra, calculus, probability, and statistics.
  • Proven track record of implementing complex deep learning algorithms from scratch.
  • Master’s or PhD in Computer Science, Machine Learning, Physics, Mathematics, or a related field, or equivalent industry experience (preferred).
  • Experience with low-level GPU programming such as CUDA or Triton, or hardware co-design (preferred).
  • Familiarity with training large language models (preferred).
  • Familiarity with inference challenges and OSS inference engines such as SGLang and vLLM (preferred).

Benefits

  • Competitive salary with a base pay range of $250,000 to $400,000 USD.
  • Meaningful equity in a fast-growing startup.
  • Comprehensive benefits package.
  • Opportunity to work on cutting-edge AI infrastructure and low-latency model serving.
  • High ownership and direct impact in a fast-growing team.
  • Work alongside world-class engineers and AI researchers.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Research Engineer (API Research)

Cato Networks 251-1K Diversified Telecommunication Services

Cato Networks is hiring a Research Engineer for its API Security Content team to analyze cloud application APIs and build content and integrations for its cloud-delivered networking and security platform.

Elasticsearch HTTP Kibana Postman Python REST API SQL
3 hours, 25 minutes ago

Environmental Research - Graduates - AI Training - Seattle, US

Prolific 51-250 Professional Services

Prolific is seeking Environmental Researchers and Sustainability Experts to help train and evaluate AI models using real-world ecological and climate expertise as part of its Expert Network.

1 day, 3 hours ago

Machine Learning Scientist

Appier 251-1K Professional Services

Appier is hiring a Machine Learning Scientist to contribute research and model development for real-world advertising and enterprise product use cases.

Machine Learning System Design
1 day, 3 hours ago

Senior Research Engineer - Experimental Engineering

Consumer Reports 251-1K Media

Consumer Reports is hiring a Senior Research Engineer to lead early-stage experimental engineering efforts focused on AI, machine learning, and emerging technologies for products like AskCR.

FastAPI Machine Learning Python React React Native
1 day, 3 hours ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers