Machine Learning Researcher / Engineer (Foundational Models) jobs in United States
cer-icon
Apply on Employer Site
company-logo

Pathway · 2 months ago

Machine Learning Researcher / Engineer (Foundational Models)

Pathway is an innovative company revolutionizing artificial intelligence with its groundbreaking post-transformer model. The role involves conducting research and development in attention-based models, focusing on model training and architecture adaptation to enhance project outcomes.

Artificial Intelligence (AI)

Responsibilities

Perform (distributed) model training
Help improve/adapt model architectures based on experiment results
Design new tasks and experiments
Optionally: oversee activities of team members involved in data preparation

Qualification

Machine Learning ResearchDeep LearningModel TrainingPyTorchGPU ArchitectureGraph AlgorithmsRespectful of othersModel MonitoringGitCI/CDFluent in English

Required

You have published at least one paper at NeurIPS, ICLR, or ICML - where you were the lead author or made significant conceptual & code contributions
You have significantly contributed to an LLM training effort which became newsworthy (topped a Huggingface benchmark, best in class model, etc.), preferably using multiple GPU's
You have spent at least 6 months working in a leading Machine Learning research center (e.g. at: Google Brain / Deepmind, Apple, Meta, Anthropic, Nvidia, MILA)
You were an ICPC World Finalist, or an IOI, IMO, or IPhO medalist in High School
A deep learning researcher, with a track record in Language Models and/or RL (candidates with a Vision or Robotics ML background are also welcome to apply)
Interested in improving foundational architectures and creating new benchmarks
Experienced at hands-on experiments and model training (PyTorch, Jax, or Tensorflow)
Have a good understanding of GPU architecture, memory design, and communication
Have a good understanding of graph algorithms
Have some familiarity with model monitoring, git, build systems, and CI/CD
Respectful of others
Fluent in English

Preferred

Knowledge of approaches used in distributed training
Familiarity with Triton
Successful track-record in algorithms & data science contests
Showing a code portfolio

Benefits

Employee Stock Option Plan
Type of contract: Full-time, permanent

Company

Pathway

twittertwittertwitter
company-logo
Shaking the foundations of AI with Dragon Hatchling (BDH), the first post-transformer model that adapts and thinks just like humans.

Funding

Current Stage
Early Stage
Total Funding
$15.5M
Key Investors
Amazon Web ServicesTQ Ventures
2025-10-08Non Equity Assistance· $1M
2024-10-09Seed· $10M
2022-12-06Pre Seed· $4.5M
Company data provided by crunchbase