Machine Learning Engineer - Generative AI

We are seeking a Machine Learning Engineer – Generative AI (3D & Real-Time Rendering) to join our onsite team in London.

In this role, you will design, implement, and optimize advanced generative AI systems for real-time photorealistic rendering and 3D applications. You will work at the intersection of deep learning and computer graphics, collaborating closely with rendering and engine teams to integrate cutting-edge AI into production-grade pipelines.

You will tackle challenging problems such as low-latency inference, controllable generation, and deploying generative models within real-time rendering 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 (diffusion, GANs, transformers).
  • Hands-on experience with modern ML frameworks (e.g., PyTorch, JAX, TensorFlow).
  • Advanced programming skills in Python.
  • Solid understanding of model training, evaluation, and scaling.
  • Strong problem-solving, analytical, and communication skills.
  • Ability to work effectively in multidisciplinary, fast-paced, research-driven teams.

Preferred Qualifications

  • Experience with 3D vision, neural rendering, or photorealistic rendering pipelines.
  • Familiarity with rendering engines (e.g., Unreal Engine, Unity) or custom 3D engines.
  • Experience with multimodal and foundation models (e.g., text-to-3D, video generation).
  • Knowledge of NeRFs, Gaussian splatting, or neural scene representations.
  • Experience optimizing models for real-time inference (TensorRT, ONNX, quantization, distillation).
  • GPU programming experience (CUDA, Triton, or similar).
  • Track record of publications at top-tier venues (NeurIPS, CVPR, ICML, ICLR, SIGGRAPH, ECCV).
  • Experience integrating ML models into production pipelines (e.g., graphics, AR/VR, simulation systems).
  • Contributions to open-source projects or collaboration with research labs.

Key Responsibilities

  • Develop and optimize state-of-the-art generative AI systems (diffusion, transformer-based models) for real-time photorealistic rendering and 3D applications.
  • Design pipelines for controllable, high-fidelity, low-latency generation within rendering systems.
  • Collaborate with graphics and rendering teams to integrate AI models into real-time engines and production pipelines.
  • Prototype, benchmark, and productionize algorithms for photorealistic 2D/3D content generation.
  • Advance techniques in neural rendering, scene generation, and AI-driven photorealism.
  • Improve model efficiency, scalability, and deployment performance.
  • Stay up to date with the latest research and rapidly experiment with emerging approaches.
  • Work cross-functionally to align technical solutions with product and business goals.

What We Offer

  • The opportunity to work at the intersection of generative AI, 3D vision, and real-time photorealistic rendering.
  • A fast-moving, research-driven environment with real product impact.
  • A culture that values technical excellence, ownership, and creativity.
  • Very attractive salary

If you are passionate about generative AI, real-time photorealistic rendering, and 3D systems, and want to help build the next generation of AI-driven graphics pipelines, we’d love to hear from you.

Job Details

Company
LIT8
Location
City of London, London, United Kingdom
Posted