Stockholm, Sweden at Encube
Core
Building AI that understands 3D models and engineering drawings to generate CAD and drawings, with a focus on model evaluation and connecting 2D drawings to 3D models.
Role type
ML Engineer and ML Ops Engineer
Builds
AI models for 3D/CAD generation, reproducible ML experiments, and production ML pipelines
Domain
Computer Vision, 3D Modeling, CAD, Generative AI
Deliverable
production ML models
Required skills
PyTorch, model training, agent development, experiment reproducibility, cost optimization, data security, GPU management
Preferred skills
3D data experience, CAD knowledge, graph neural networks, SkyPilot, MLflow
Technologies
PyTorch, SkyPilot, MLflow, AWS, on-prem GPUs
Responsibilities
Train models connecting 2D drawings to 3D models; build agents that generate CAD and drawings; manage on-prem and AWS GPU infrastructure; ensure experiment reproducibility and data safety; optimize ML team workflow and costs
Seniority
Mid-level, hands-on IC
