PhD Intern - Machine Learning

2 months ago
Internship
In-person · Fremont, CA, US... more

About Pacific Fusion 

Pacific Fusion was founded in 2023 with the mission to power the world with abundant, affordable, clean energy.

We are rapidly designing and building a pulsed magnetic fusion system to achieve net facility gain (more fusion energy output than stored energy input). In parallel, we are developing the key components required to build affordable fusion systems.

Our plans are ambitious. We have raised over $900M so far from incredible investors, resourcing us to deliver on these plans. We are bringing together the best scientists, engineers, and operators from the fusion community, hard tech industry, and other sectors. We are united by a shared sense of urgency to provide clean power for the world, particularly managing climate change while meeting growing global energy demand. You can read more about us in this letter from our founders.

About the Role

Job Overview:

Pacific Fusion is seeking an exceptional research intern to join our team and create the path to commercial fusion energy.

At Pacific Fusion, the machine learning team works across several computational teams to accelerate and improve our critical computational capabilities. In this role, you will contribute to the development of machine learning systems that make our multi-physics simulators faster, higher fidelity, and overall more impactful to their internal users. You will collaborate with computational physicists, target designers, and experimental physicists to find the highest leverage research work that enables us to move faster and more efficiently.

We are looking for candidates interested in a 16 to 24 week, paid, in-person internship based in our Fremont, CA office. 

Responsibilities:

The main responsibility of a research intern in this role is to carry out research towards developing novel ML methods for accelerating expensive CFD-like simulations used for fusion target design. You will be expected to (1) read through ML and physics literature that is relevant to your project, (2) design, implement and train novel ML architectures, and (3) produce internal and/or external reports of your work in the form of whitepapers or publications to top-tier research venues.

Strong candidates will have the following core competencies:

  • ML and physics expertise: research interns in this role work at the intersection of machine learning and computational physics, and must be able to reason across both domains (although it is fine to have deeper background in one or the other).

  • Communication: you’ll be working at the intersection of multiple highly technical disciplines, and you will collaborate with a diverse set of subject matter experts across computational, theoretical, and experimental plasma physics. You will be expected to work with your colleagues to translate these problems into machine learning ones.

  • Agency: we have a lot of exciting machine learning problems at Pacific Fusion. You’ll work in an iterative, kind, and deeply collaborative environment to new ML capabilities in computational plasma physics. You’ll be given the freedom as well as the responsibility to push directions that you believe in and that have the chance to provide the greatest internal and external impact.

Qualifications:

  • Currently in the process of obtaining a PhD in Machine Learning, computational physics, plasma physics, or related fields.

  • Hands-on and mature experience in deep learning algorithms, techniques, and frameworks such as PyTorch and JAX.

  • Software development knowledge and experience in Python and the relevant scientific and computational stack (numpy, scipy, matplotlib, etc).

  • Experience on applying machine learning to model interesting physical problems.

  • An interest in pushing the state of the art on problems at the intersection of physics and machine learning, including building surrogate models of physical systems, learning and optimizing PDEs, unsupervised learning for physical data, automatic differentiation, bayesian inference, and so on.

  • Proven track record of publications at ML venues such as NeurIPS, ICML, ICLR, TMLR, and/or in top-tier physics journals.

Actual base salary offered will be determined by: experience, skills, and work location. This range is for base salary, our total compensation includes equity and benefits. We welcome you to apply even if your expectations are outside our listed range. 

Pacific Fusion is committed to creating a diverse environment and is proud to be an Equal Opportunity Employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, gender, gender identity or expression, sexual orientation, national origin, genetics, disability, age, or veteran status.

Pacific Fusion is committed to providing reasonable accommodations for qualified individuals with disabilities in our job application procedures and throughout employment. If you need assistance or any accommodation, please let us know. 

Pacific Fusion does not accept unsolicited resumes from recruiters or employment agencies without a fully executed recruitment agreement in place. In the absence of such agreement, Pacific Fusion reserves the right to pursue and hire any candidates without an obligation to pay fees. Agencies are requested not to contact Pacific Fusion hiring managers or employees regarding recruiting services.