Machine Learning Engineer - Generative AI

Company Description

We are a startup building next-generation real-time image-to-image transformation models, with a special focus on 3D applications and rendering engine integration. Leveraging the latest in GANs, diffusion models, and large-scale deep learning, our research-driven team values autonomy, creativity, and technical excellence. Join us to help shape the future of real-time 2D/3D generative AI in a highly collaborative and innovative environment.

Role Description

We are seeking a Machine Learning Engineer – Generative AI to join our onsite team in London. In this role, you will design, implement, and optimize advanced generative AI models for real-time image and 3D applications, collaborating closely with experts in computer graphics and rendering. You’ll have the opportunity to work on some of the most technically challenging problems at the intersection of deep learning and real-time 3D systems.

Minimum Qualifications

• Ph.D. in Computer Science, Mathematics, Electrical Engineering, or a related field; or Master’s degree with 2+ years of relevant industry experience; or Bachelor’s degree with 4+ years of relevant industry experience.
• Strong expertise in deep learning, neural networks, and generative models (GANs, diffusion models).
• Practical experience with modern machine learning frameworks (e.g., PyTorch, TensorFlow).
• Advanced programming skills in Python.
• Strong problem-solving, analytical, and communication skills.
• Demonstrated ability to work effectively in multidisciplinary, fast-paced, research-driven teams.

Preferred Qualifications

• Experience with 3D vision, computer graphics, or real-time rendering engines (e.g., Unreal Engine, Unity, custom 3D engines).
• Proven track record of publications at top-tier conferences (e.g., NeurIPS, CVPR, ICML, ICLR, SIGGRAPH, ECCV).
• Experience with GPU programming (CUDA) and model optimization for real-time inference (e.g., quantization, pruning, ONNX, TensorRT, custom CUDA kernels).
• Background in scalable algorithm design for real-time or interactive applications.
• Experience integrating machine learning models with complex production pipelines, including 3D graphics or AR/VR systems.
• Contributions to open-source research codebases or prior collaboration with academic/industry research labs.

Key Responsibilities

• Develop and optimize state-of-the-art generative models (GANs, diffusion models) for real-time image-to-image and 3D/graphics tasks.
• Collaborate with 3D graphics and rendering teams to integrate AI models into interactive applications and pipelines.
• Prototype, benchmark, and productionize new algorithms that advance real-time generative AI for both 2D and 3D content.
• Research and experiment with emerging techniques to keep our technology at the cutting edge.
• Work cross-functionally to align technical solutions with business goals and product needs.

What We Offer

• The opportunity to work at the intersection of deep learning, 3D vision, and real-time graphics.
• A dynamic, research-driven environment where your work directly impacts next-gen 3D/AI products.
• Access to world-class compute resources and support for continued research and publication.
• A culture that values technical excellence, creativity, and personal growth.
• Competitive salary.

If you are passionate about generative AI, love working with 3D graphics and rendering technology, and want to be part of a world-class team building the future of real-time AI, we’d love to hear from you!