Applied Scientist, Tabular Foundational Model, AWS

Amazon Science; Seattle (WA); On-site; USD 167,000 - 226,000; Full time

Benefits offered by this job

Health insurance

401(k) matching

Paid time off

Parental leave

Job summary

Amazon Science is hiring researchers to advance Tabular Foundational Models (TFM) as part of the Mitra family, enabling zero-shot analytics and automated feature engineering on enterprise tabular data.

You will collaborate with world-class researchers and engineers, publish findings, and apply scalable ML solutions across real-world business problems in AWS AI, leveraging vast datasets and compute resources.

Qualifications

• 3+ years of building ML models for business applications.
• PhD, or Master’s with 6+ years of applied research.
• Experience programming in Java, C++, Python or related language.
• Experience with neural deep learning methods and ML.

Responsibilities

• Develop and advance Tabular Foundational Models (TFM) for enterprise data.
• Collaborate with researchers and engineers to publish findings.
• Partner with business teams to apply scalable ML solutions.

Skills

ML model development

Deep learning

Programming: Java/C++/Python

Research experience in ML

Education

PhD or Master's with 6+ years research

Tools

R

scikit-learn

Spark

MxNet

TensorFlow

Job description

Description

Do you want to work on Tabular Foundational Model (TFM) to evolutionize how enterprises unlock insights from structured data? We're building Amazon's frontier large-scale tabular foundation model (Mitra family of models) that understands tables natively—enabling zero-shot analytics, automated feature engineering, and cross-domain generalization across diverse tabular datasets.

Description

Do you want to work on Tabular Foundational Model (TFM) to evolutionize how enterprises unlock insights from structured data? We're building Amazon's frontier large-scale tabular foundation model (Mitra family of models) that understands tables natively—enabling zero-shot analytics, automated feature engineering, and cross-domain generalization across diverse tabular datasets.
Come join the world class researchers and academics in the AWS AI endeavor, and develop the science that powers countless new businesses in cloud computing!
AWS, the world-leading provider of cloud services. Our customers bring problems that will give Applied Scientists like you endless opportunities to see your research have a positive and immediate impact in the world. You will have the opportunity to partner with technology and business teams to solve real-world problems, have access to virtually endless data and computational resources, and to world-class engineers and developers that can help bring your ideas into the world. As part of the team, we expect that you will develop innovative solutions to hard problems, and publish your findings at peer reviewed conferences and journals.
The scientific topics you are going to work on include, but are not limited to: pre-training tabular foundational models (Mitra family of models) on heterogeneous tabular data, few-shot adaptation for domain-specific tasks (regression, classification), agentic feature engineering for handling high-cardinality features and missing data patterns, etc.

Basic Qualifications

• 3+ years of building machine learning models for business application experience
• PhD, or Master's degree and 6+ years of applied research experience
• Experience programming in Java, C++, Python or related language
• Experience with neural deep learning methods and machine learning
  

Preferred Qualifications

• Experience with modeling tools such as R, scikit-learn, Spark MLLib, MxNet, Tensorflow, numpy, scipy etc.
• Experience with large scale distributed systems such as Hadoop, Spark etc.
  

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.
Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.
The base salary range for this position is listed below. Your Amazon package will include sign-on payments and restricted stock units (RSUs). Final compensation will be determined based on factors including experience, qualifications, and location. Amazon also offers comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage), 401(k) matching, paid time off, and parental leave. Learn more about our benefits at https://amazon.jobs/en/benefits.
USA, WA, Seattle - 167,100.00 - 226,100.00 USD annually