← Back to jobs

Applied ML Graph Neural Network Research Software Engineer

  • Zürich, Switzerland

About the role

Minimum qualifications

  • Bachelor’s degree or equivalent practical experience.
  • 5 years of experience with ML infrastructure (e.g., model deployment, model evaluation, optimization, data processing, debugging).
  • Experience designing, training, and deploying Graph Neural Networks (GNNs) or graph embedding models using frameworks such as PyTorch Geometric (PyG), JAX/Jraph, or TF-GNN.