Job 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. - 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 - 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 amazon.jobs 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 amazon.jobs . USA, WA, Seattle - 167,100.00 - 226,100.00 USD annually

Requirements

Must-haves

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!
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
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.
USA, WA, Seattle - 167,100.00 - 226,100.00 USD annually

Nice-to-haves

None flagged.

Application prep

Use the role’s language and application details while tailoring your materials.

Keywords to mirror

awsaimachine learningjavapython

Platform: amazon
Effort: medium

Benefits

Benefits detected in the listing.

Health Insurance

Dental

Vision

401(K)

Paid Time Off

Parental Leave