Ai Hardware Design Engineer

Ai Hardware Design Engineer

Lead I - Data Science

Who We Are:

Born digital, UST transforms lives through the power of technology. We walk alongside our clients and partners, embedding innovation and agility into everything they do. We help them create transformative experiences and human-centered solutions for a better world.

UST is a mission-driven group of 29,000+ practical problem solvers and creative thinkers in more than 30 countries. Our entrepreneurial teams are empowered to innovate, act nimbly, and create a lasting and sustainable impact for our clients, their customers, and the communities in which we live.

With us, you'll create a boundless impact that transforms your career-and the lives of people across the world.

Visit us at UST.com.

You Are:

We are seeking an Ai Hardware Design Engineer to join our team and drive innovation in AI-powered solutions. This role involves designing, developing, and optimizing generative AI models and workflows for applications such as content creation, product design, and intelligent automation.

The opportunity:

* Develop forward surrogate models for CVD/ALD/etch chambers mapping geometry, gas chemistry, flow, temperature, and power to film-uniformity, step-coverage, particle behavior, and thermal outcomes.

* Implement inverse-design workflows where target performance specifications generate feasible chamber geometries, showerhead/baffle designs, and process conditions via generative or adjoint/topology-optimization methods.

* Build bi-directional models that infer optimal process parameters for a given geometry and recommend geometry modifications when process latitude is insufficient.

* Create high-fidelity digital twins combining physics-based solvers (CFD, plasma, heat transfer) with learned surrogate components for rapid design-space exploration.

* Platform & MLOps Infrastructure: Implement and maintain robust, containerized MLOps systems (Docker, Kubernetes) in HPC environments to deploy models efficiently.

* Develop robust multi-objective optimization and uncertainty-quantification workflows to ensure AI-generated designs are manufacturable, robust to variation, and compatible with downstream yield requirements.

This position description identifies the responsibilities and tasks typically associated with the performance of the position. Other relevant essential functions may be required.

What you need:

* Education: Master's or Ph.D. in Computer Science, Computational/Electrical Engineering, AI/ML, or related field.

* Technical Expertise:

* Strong proficiency in Python and ML frameworks (PyTorch, TensorFlow).

* Experience with generative AI (LLMs, diffusion models, graph-based models).

* Knowledge of computational materials methods (DFT, MD, phase-field modeling).

* Additional Skills:

* Familiarity with MLOps, HPC environments, and cloud deployment.

* Proven experience (code repos, publications) bridging simulation software, hardware design, and ML.

Compensation can differ depending on factors including but not limited to the specific office location, role, skill set, education, and level of experience. UST provides a reasonable range of compensation for roles that may be hired in various U.S. markets as set forth below.

Role Location: California

Compensation Range : $86,000-$129,000

Benefits

Full-time, regular employees accrue a minimum of 10 days of paid vacation per year, receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year), 10 paid holidays, and are eligible for paid bereavement leave and jury duty. They are eligible to participate in the Company's 401(k) Retirement Plan with employer matching. They and their dependents residing in the US are eligible for medical, dental, and vision insurance, as well as the following Company-paid Employee Only benefits: basic life insurance, accidental death and disability insurance, and short- and long-term disability benefits. Regular employees may purchase additional voluntary short-term disability benefits, and participate in a Health Savings Account (HSA) as well as a Flexible Spending Account (FSA) for healthcare, dependent child care, and/or commuting expenses as allowable under IRS guidelines. Benefits offerings vary in Puerto Rico.

Part-time employees receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year) and are eligible to participate in the Company's 401(k) Retirement Plan with employer matching.

Full-time temporary employees receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year) and are eligible to participate in the Company's 401(k) program with employer matching. They and their dependents residing in the US are eligible for medical, dental, and vision insurance.

Part-time temporary employees receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year).

All US employees who work in a state or locality with more generous paid sick leave benefits than specified here will receive the benefit of those sick leave laws.

What we believe:

We proudly embrace the values that have shaped UST since day one. We build our culture of Humility, Humanity, and Integrity. These values inspire us to nurture a people-first, human centric culture that fosters diversity, prioritizes sustainable solutions, and keeps our people and clients at the forefront of all decisions.

Humility:

We will listen, learn, be empathetic and help selflessly in our interactions with everyone.

Humanity:

Through business, we will better the lives of those less fortunate than ourselves.

Integrity:

We honor our commitments and act with responsibility in all our relationships.

Equal Employment Opportunity Statement

UST is an Equal Opportunity Employer.

All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, status as a protected veteran, or any other applicable characteristics protected by law. We will consider qualified applicants with arrest or conviction records in accordance with state and local laws and "fair chance" ordinances.

UST reserves the right to periodically redefine your roles and responsibilities based on the requirements of the organization and/or your performance.

#UST

#LI-MK2

machine learning,ai hardware design,generative ai,python,