Home > Vacancies > ML Engineer (Physics-Informed / GNN)  

ML Engineer (Physics-Informed / GNN)  

Level

from Middle

Specialization

GNN, PINNs, PyTorch, TensorFlow, DGL, Python, MLOps, NumPy, SciPy, Pandas, GraphSAGE, etc.

Location

All Locations

About this position

SoftTeco is seeking an AI/ML Engineer (with a focus on Physical Systems) to develop advanced machine learning models for the oil and gas industry.

Project Focus: You will design and implement Graph Neural Networks (GNNs) that incorporate physical laws (Physics-Informed Neural Networks) to build high-precision surrogate models. These models will predict the behavior of complex engineering systems (e.g., pressure distribution and flow rates in pipeline networks) using data derived from physics-based simulations.

Responsibilities

  • Design and implement Neural Network architectures to model flow dynamics in interconnected pipeline networks;
  • Build surrogate models that accurately predict pressure distributions, flow rates, and network behavior under varying operational scenarios (training data is acquired through running simulations of the physics models);
  • Create data pipelines to extract network topology and simulation results from physics-based models (Nexus/Prosper) and transform them into graph representations;
  • Develop training frameworks that incorporate physics constraints (conservation laws, pressure-flow relationships) into neural network loss functions;
  • Collaborate with petroleum engineers to ensure model predictions align with physical behavior and operational constraints;
  • Implement model monitoring, validation, and continuous improvement workflows.

Technical skills

  • 5+ years of experience in a relevant role, such as ML Engineer, GNN Specialist, or Research Scientist specializing in Graph ML and surrogate modeling;
  • Expertise in Graph Neural Networks (e.g., GCN, GraphSAGE, Message Passing) and proven implementation experience;
  • Proficiency with deep learning frameworks (PyTorch Geometric, DGL, or TensorFlow GNN);
  • Experience in building surrogate models or Physics-Informed Neural Networks (PINNs) for engineering applications;
  • Strong Python programming skills and experience with scientific computing libraries (NumPy, SciPy, Pandas);
  • Experience working with complex data structures (graphs, time series, spatial data);
  • Understanding of optimization techniques and experience in handling large-scale training datasets;
  • Experience with hyperparameter tuning, model building, and uncertainty quantification.

Non-technical skills

  • English proficiency at a B2 level or higher;
  • Strong problem-solving skills and a research-oriented mindset;
  • Ability to collaborate effectively with petroleum engineers to validate, refine, and align models;
  • A strong motivation for deep, long-term engagement in solving complex and meaningful problems that bridge scientific research and industrial engineering.

Work schedule and format

  • Full-time position with a flexible schedule (Monday to Friday);
  • This is a work-from-anywhere position. All you need is a reliable Internet connection;
  • On-site onboarding: Must be available to complete a one-month project immersion in Kuwait (all travel and accommodation expenses covered).

Interview process

  • Interview with a Recruiter & English Check (30 min);
  • Technical interview covering relevant skills with SoftTeco’s experts (1 hour);
  • Client-side interview in English;
  • Final HR interview to discuss work conditions and salary.

Feel like you are a great match this position? Explore your next career opportunity at SoftTeco!

Our Benefits

Comfortable working environment & flexible hours; opportunity for remote work.

Holidays based on the official calendar of your country.

Relocation opportunities or options to work from SoftTeco’s offices based in other countries.

Health insurance, compensation packages for sports; and corporate discounts from SoftTeco’s partners.

Mentoring, internal knowledge sharing, all-hands meetings, and our corporate English courses.

Comprehensive Career Pathway system designed to facilitate employees’ professional growth and development within the organization.

Corporate events for employees and their families, sponsored by the company; gifts on working anniversaries and professional holidays.

Working at SoftTeco

We understand how important it is to find a vacancy that would fit your tech stack and skills level. But we also understand that it’s even more important to find a team of people who will make your work comfortable and enjoyable, and who will inspire you to grow and learn. Trust, expertise, rationality, and involvement are the core qualities that form SoftTeco’s culture and help us build long-term relationships with our employees and partners.

Current Job Openings

SoftTeco constantly expands its expertise and as we do so, we are in need of top industry talents to help us resolve complex technical challenges and bring innovation to our clients. If you are looking for the next step in your career, we encourage you to explore our Job Openings page and see if your perfect vacancy awaits you.

    Ready to join the team?

    Submit your CV via the form below and we will consider you for the future open positions that match your skill set and experience.

    I agree with the Privacy Policy and the Terms of Services

    Softteco Logo Footer