.
AI/ML Engineer with GNN & ScaNN
  • Warsaw
AI/ML Engineer with GNN & ScaNN
Warszawa, Warsaw, Masovian Voivodeship, Polska
IN4GE sp. z o.o.
24. 7. 2025
Informacje o stanowisku

technologies-expected :


  • Google Cloud Platform
  • TensorFlow
  • PyTorch
  • JAX
  • Hugging Face Transformers

about-project :


  • We’re looking for an experienced AI/ML Engineer to join a team focused on building cutting-edge solutions using Generative AI, Graph Neural Networks (GNN), and scalable data pipelines on Google Cloud Platform. In this role, youll have the opportunity to design and implement advanced machine learning models (LLMs, diffusion models), work with real-world data at scale, and deploy solutions in production using modern MLOps tools and practices.

responsibilities :


  • Design, implement, and optimize Graph Neural Network (GNN) architectures.
  • Build and maintain embedding generation pipelines using ScaNN and related tools.
  • Develop and train advanced machine learning models, including LLMs and diffusion models.
  • Work with large-scale datasets and build robust data pipelines using GCP services (BigQuery, Dataflow).
  • Leverage Vertex AI (Workbench, Training, Prediction, Pipelines, Feature Store) for end-to-end ML workflows.
  • Apply MLOps best practices to deploy, monitor, and manage models in production environments.
  • Collaborate with cross-functional teams including data engineering, research, and software engineering.

requirements-expected :


  • Bachelor’s degree in Computer Science, Machine Learning, Artificial Intelligence, or a related field - or equivalent practical experience.
  • 5+ years of experience in AI/ML engineering, including at least 2+ years working with Generative AI (e.g., LLMs, diffusion models).
  • Hands-on experience with GCP AI/ML services, particularly Vertex AI components.
  • Proficiency in Python and strong knowledge of ML frameworks such as TensorFlow, PyTorch, JAX, and Hugging Face Transformers.
  • Experience with data engineering and building scalable AI/ML data pipelines on GCP.
  • Solid foundation in machine learning algorithms, deep learning, neural networks, and NLP.
  • Familiarity with MLOps tools and principles for model lifecycle management.
  • Strong analytical and problem-solving skills, with a drive for building impactful AI systems.
  • Fluent English (C1) and ability to work effectively in a distributed, international team.
  • Certification as a PMLE (Professional Machine Learning Engineer) or equivalent qualification is required.

offered :


  • Fully remote work with flexible working hours - EMEA Timezone.
  • Long-term collaboration on B2B contract.
  • Opportunity to work on complex cloud projects for international clients.
  • Professional growth in a highly skilled and supportive team.
  • Collaborative and open working culture.

benefits :


  • remote work opportunities
  • flexible working time
  • integration events
  • no dress code

  • Praca Warszawa
  • Warszawa - Oferty pracy w okolicznych lokalizacjach


    84 382
    7 517