Skip to main contentA logo with &quat;the muse&quat; in dark blue text.

AI/ML GPU Software Engineer

AT RWE Clean Energy
RWE Clean Energy

AI/ML GPU Software Engineer

Bellevue, WA

RWE Supply & Trading Americas, LLC
To start as soon as possible, fulltime, permanent


Your future plans

We are building our core AI Lab team to explore creative applications of AI methods to challenges in the renewable energy industry. To help us innovate and build industry-leading AI models, we are seeking a highly motivated software engineer with expertise in leveraging NVIDIA GPUs for machine learning applications.

As a key member of our engineering team, you will be responsible for designing, developing, and optimizing GPU-accelerated software solutions that enable the training, inference, and deployment of machine learning models at scale. The successful candidate will play a critical role in developing and maintaining the tech stack of the lab and will work with senior scientists and engineers from a variety of fields. RWE is the third largest producer of wind power in the world, so you will have a material impact on renewable energy production across the globe. We operate a hybrid model that requires employees to be in the office at least three days per week to foster collaboration and teamwork, while offering the flexibility to work remotely as applicable.

Want more jobs like this?

Get Data and Analytics jobs in Bellevue, WA delivered to your inbox every week.

By signing up, you agree to our Terms of Service & Privacy Policy.


If you're passionate about leveraging GPUs for state-of-the-art production machine learning models and thrive in a fast-paced, collaborative environment, we'd love to hear from you! Your responsibilities will include:

  • Design and implement GPU-accelerated algorithms and software frameworks for training and inference of machine learning models.
  • Optimize deep learning workflows for execution on NVIDIA GPUs using frameworks such as TensorFlow, PyTorch, or MXNet.
  • Develop custom CUDA kernels and GPU-accelerated libraries to accelerate performance-critical operations in machine learning pipelines.
  • Collaborate with data scientists, machine learning engineers, and software developers to integrate GPU-accelerated components into end-to-end machine learning systems
  • Stay up to date with the latest advancements in NVIDIA GPU architectures, deep learning frameworks, and GPU-accelerated libraries
  • Contribute to a team culture where diverse viewpoints, backgrounds and expertise are welcomed

Your powerful skills

  • 3+ years experience in software engineering with a focus on leveraging NVIDIA GPUs for machine learning
  • Proficiency in Nvidia GPU frameworks (CUDA, cuDNN, TensorRT, NCCL) and HPC technologies
  • Understanding of distributed training techniques and frameworks for scaling machine learning workflows across multiple GPUs or nodes
  • Proficiency in programming languages such as Python and Julia, and experience with machine learning packages (e.g., PyTorch, TensorFlow, Jax)
  • Deep expertise in GPU hardware, architecture, and performance optimization
  • Proficiency in GPU profiling and debugging tools such as Nsight and nvprof
  • Skilled in container technologies and orchestration (Docker, Kubernetes) optimized for GPU workflows
  • Excellent problem-solving skills, attention to detail, and a passion for driving innovation in state of the art machine learning models
  • Strong communication and collaboration skills, with the ability to work effectively in cross-functional teams and contribute to a positive team climate

Advantageous, but not a must

  • Experience with stream processing frameworks (e.g., Apache Kafka, Apache Flink) for real-time data ingestion and processing
  • Experience with Google Cloud Platform MLOps tools (e.g., Vertex AI)
  • Knowledge of serverless computing and event-driven architectures for building scalable and cost-effective pipelines
  • Familiarity with monitoring, logging, and debugging tools for distributed systems (e.g., Prometheus, ELK Stack)
  • Experience with applications of AI/ML tools in the energy, weather, or climate fields

What we value most is passion, collaboration, diversity of backgrounds and expertise, willingness to learn and innovate, and a determined and resilient work ethic. Even if you think you do not yet display all the skills above, we would still like to hear from you.

Our offer

We really appreciate you going the extra mile and using every ounce of energy when the heat is on. That's why we want to make your time with us as enjoyable as possible. At RWE Supply & Trading we value our employees and strive to create a supportive and inclusive work environment. With a range of competitive benefits, we not only offer excellent development prospects and an attractive remuneration package, but also take care of your well-being and understand that life-work integration is essential. Also our offer includes the following perks:

  • Task oriented and hybrid working model
  • Diverse and multicultural team in a highly dynamic and rapidly growing business
  • Relocation expenses for moving to the greater Seattle area if you are not already local
  • Salary: $120,000 - $170,000

rwe.com/career

Any questions? Patrycja Bartela (Recruiting), E: Patrycja.Bartela@rwe.com

Apply now with just a few clicks: ad code 87232

We look forward to meeting you!

We value diversity and therefore welcome all applications, irrespective of sex, disability, nationality, ethnic and social background, religion and beliefs, age or sexual orientation and identity. Of course, you can find us on LinkedIn, Facebook, Twitter and Xing, too.

RWE Supply & Trading is the interface between RWE and the energy markets around the world. Around 2,000 employees from over 70 different countries trade electricity, gas, commodities and CO2 emission allowances. The trading entity also ensures the commercial optimisation of RWE's power plant dispatch and markets electricity from RWE.

Client-provided location(s): Bellevue, WA, USA
Job ID: RWE_Renewable-87232-en_GB
Employment Type: Other

Perks and Benefits

  • Health and Wellness

    • Health Insurance
    • Dental Insurance
    • Vision Insurance
    • Life Insurance
    • Short-Term Disability
    • Long-Term Disability
    • FSA
    • FSA With Employer Contribution
    • HSA With Employer Contribution
    • HSA
    • On-Site Gym
  • Parental Benefits

    • Adoption Leave
    • Birth Parent or Maternity Leave
    • Non-Birth Parent or Paternity Leave
    • Family Support Resources
    • Fertility Benefits
    • Adoption Assistance Program
  • Work Flexibility

    • Hybrid Work Opportunities
  • Office Life and Perks

    • Commuter Benefits Program
    • Snacks
    • Holiday Events
    • Casual Dress
  • Vacation and Time Off

    • Paid Vacation
    • Paid Holidays
    • Personal/Sick Days
    • Leave of Absence
  • Financial and Retirement

    • 401(K) With Company Matching
    • 401(K)
    • Relocation Assistance
    • Profit Sharing
  • Professional Development

    • Tuition Reimbursement
    • Access to Online Courses
    • Internship Program
    • Lunch and Learns
  • Diversity and Inclusion

    • Unconscious Bias Training
    • Employee Resource Groups (ERG)
    • Diversity, Equity, and Inclusion Program