Applied Scientist, Tabular Foundational Model, AWS

Amazon Inc.; Omaha, Northern (NE, KY); Hybrid; 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 is seeking an Applied Scientist to advance Tabular Foundational Models (TFM) at AWS. You will work on pre-training models for heterogeneous tabular data, enabling zero-shot analytics and automated feature engineering, with opportunities to publish and contribute to cutting-edge research.

Join a world-class research team, partner with tech and business units, and impact cloud computing by delivering scalable ML solutions and novel modeling techniques for enterprise data.

Qualifications

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

Responsibilities

• Develop tabular foundational models and few-shot adaptations for diverse datasets.
• Collaborate with cross-functional teams to apply research to real-world data problems.
• Publish findings at peer-reviewed conferences and journals.

Skills

Java

C++

Python

Machine learning

Education

PhD or MS + 6 years

Tools

R

scikit-learn

Spark MLLib

TensorFlow

Job description

Applied Scientist, Tabular Foundational Model, AWS

Job ID: 10560684 | Amazon Development Center U.S., Inc.

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

Important FAQs for current Government employees

Before proceeding, please review the following FAQs

https://amazon.jobs/en/faqs#faqs-for-us-government-employees

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status. Veterans, military spouses, and people with disabilities are encouraged to apply.