Taro Logo

Machine Learning Researcher / Engineer (Foundational Models)

Pathway is an enabler for Live AI, allowing organizations to run contextualized ML models connected to ever-changing enterprise data.
$100,000 - $250,000
Machine Learning
Senior Software Engineer
Remote
11 - 50 Employees
5+ years of experience
AI · Enterprise SaaS

Description For Machine Learning Researcher / Engineer (Foundational Models)

Pathway, a VC-funded AI infrastructure provider, is seeking a Machine Learning Researcher/Engineer for an ambitious foundational project in attention-based models. The company, which recently raised $10M+ in seed funding, enables Live AI for organizations to run contextualized ML models with ever-changing enterprise data. This R&D position focuses on foundational model research with a guaranteed 7-digit GPU budget.

The role combines cutting-edge research in language models, reinforcement learning, and foundational architectures with practical implementation. The ideal candidate should have a strong publication record or significant contributions to LLM training efforts. The position offers the opportunity to work on pioneering "Live AI" challenges around long sequences and changing data.

The company's impressive background includes collaboration with notable figures in AI, such as Lukasz Kaiser (co-inventor of Transformers), and leadership with connections to Geoff Hinton and Yoshua Bengio. Their client portfolio spans mobility, IoT data, logs and transactions, including major organizations like NATO.

This position offers competitive compensation including a six-digit salary and equity, with the flexibility to work remotely or from offices in Menlo Park, Paris, or Wroclaw. The role provides an opportunity to make significant impact in advancing the state-of-the-art in AI while working with a team of accomplished researchers and engineers.

Last updated 2 months ago

Responsibilities For Machine Learning Researcher / Engineer (Foundational Models)

  • Perform distributed model training
  • Improve/adapt model architectures based on experiment results
  • Design new tasks and experiments
  • Oversee activities of team members involved in data preparation

Requirements For Machine Learning Researcher / Engineer (Foundational Models)

Python
  • Published at least one paper at NeurIPS, ICLR, or ICML as lead author or significant contributor
  • Contributed to a newsworthy LLM training effort
  • 6 months experience in leading Machine Learning research center
  • ICPC World Finalist, or IOI, IMO, or IPhO medalist
  • Deep learning research background
  • Experience with PyTorch, Jax, or Tensorflow
  • Understanding of GPU architecture, memory design, and communication
  • Understanding of graph algorithms
  • Familiarity with model monitoring, git, build systems, and CI/CD
  • Fluent in English

Benefits For Machine Learning Researcher / Engineer (Foundational Models)

Equity
  • Full-time, permanent contract
  • Six-digit annual salary
  • Employee Stock Option Plan
  • Remote work option
  • Flexible office locations in Menlo Park, Paris, or Wroclaw

Interested in this job?

Jobs Related To Pathway Machine Learning Researcher / Engineer (Foundational Models)