Research Scientist - 3D Diffusion

SpAItial

SpAItial is pioneering the next generation of World Models, pushing the boundaries of generative AI, computer vision, and the simulation of reality. We are moving beyond 2D pixels to build models that natively understand the physics and geometry of our world. Our mission is to redefine how industries, from robotics and AR/VR to gaming and cinema, generate and interact with physically-grounded 3D environments. We’re looking for individuals who are bold, innovative, and driven by a passion for pushing the boundaries of what’s possible. You should thrive in an environment where creativity meets challenge and be fearless in tackling complex problems. Our team is built on a foundation of dedication and a shared commitment to excellence, so we value people who take immense pride in their work and place the collective goals of the team above personal ambition. As a part of SpAItial, you’ll be at the forefront of the AI revolution in generative AI technology, and we want you to be excited about shaping the future of this dynamic field. If you’re ready to make an impact, embrace the unknown, and collaborate with a talented group of visionaries, we want to hear from you. Responsibilities * Architect generative world models that reason about space, time, and physics. * Design and develop image/video diffusion foundational ML models. * Large-scale distributed model training on top of distributed cloud infra. * Develop demos showcasing the trained model prototypes. * Processing and maintaining large data for model training. * Productionizing models, test-time model optimization. Key Qualifications: * PhD in Computer Science or related field with a top-tier publication record in ML, Vision or graphics (CVPR, ECCV/ICCV, NeurIPS, SIGGRAPH, etc.). * Strong knowledge of generative models such as image or video diffusion. * Strong knowledge of cutting-edge architectures including diffusion transformers (DiT). * Rich experience with deep learning frameworks such as PyTorch. * Expert coding skills, and ability to rapidly iterate through ML experiments, including usage of modern AI coding tools. At SpAItial, we are committed to creating a diverse and inclusive workplace. We welcome applications from people of all backgrounds, experiences, and perspectives. We are an equal opportunity employer and ensure all candidates are treated fairly throughout the recruitment process.

PhysicsX

ABOUT US PhysicsX is a deep-tech company with roots in numerical physics and Formula One, dedicated to accelerating hardware innovation at the speed of software. We are building an AI-driven simulation software stack for engineering and manufacturing across advanced industries. By enabling high-fidelity, multi-physics simulation through AI inference across the entire engineering lifecycle, PhysicsX unlocks new levels of optimization and automation in design, manufacturing, and operations — empowering engineers to push the boundaries of possibility. Our customers include leading innovators in Aerospace & Defense, Materials, Energy, Semiconductors, and Automotive. NOTE: WE ARE CURRENTLY RECRUITING FOR MULTIPLE LEVELS AND POSITIONS, HOWEVER PLEASE ONLY APPLY FOR THE ROLE THAT BEST ALIGNS WITH YOUR SKILLSET AND CAREER GOALS. WHAT YOU WILL DO  * Work closely with our machine learning engineers, simulation engineers, and customers to translate physics and engineering challenges into mathematical problem formulations. * Build models to predict the behaviour of physical systems using state-of-the-art machine learning and deep learning techniques. * Own Research work-streams at different levels, depending on seniority. * Discuss the results and implications of your work with colleagues and customers, especially how these results can address real-world problems. * Collaborate with colleagues beyond the research team to translate your models into production-ready code. * Communicate your work to others internally and externally as called for in paper publication venues, industry workshops, customer conversations, etc. This will involve writing for academic and non-academic audiences. * Foster a nurturing environment for colleagues with less experience in DS / ML / Stats for them to grow and you to mentor.   WHAT YOU BRING TO THE TABLE * Enthusiasm about using machine learning, especially deep learning and/or probabilistic methods, for science and engineering. * Ability to scope and effectively deliver projects. * Strong problem-solving skills and the ability to analyse issues, identify causes, and recommend solutions quickly. * Excellent collaboration and communication skills — with teams and customers alike. * PhD in computer science, machine learning, applied statistics, mathematics, physics, engineering, or a related field, with particular expertise in any of the following: 1. operator learning (neural operators), or other probabilistic methods for PDEs; 2. geometric deep learning or other 3D computer vision methods for point-cloud or mesh-structured data; 3. generative models for geometry and spatiotemporal data (VAEs, Diffusion Models, Bayesian non-parametric, scaling to large datasets, etc.). * >2 years of experience in a data-driven role in a professional industry setting (excluding post-doc positions), with exposure to: * building machine learning models and pipelines in Python, using common libraries and frameworks (e.g., NumPy, SciPy, Pandas, PyTorch, JAX), especially including deep learning applications; * developing models for bespoke problem settings that involve high-dimensional data (spatiotemporal, geometric, physical); * iterating on network architectures and model structure, tuning and optimising for inductive biases, improved generalisability, and improved performance; * combining theoretical reasoning with empirical intuition to guide investigation; * formulating and running experiment pipelines to benchmark models and produce comparable results; * writing skills for communication complex technical concepts to peers and non-peers, tailoring the message for the required audience. * Publication record in reputable venues that demonstrates mastery in your field, and in particular the domains of interest listed above. Desirable venues include (but not limited to): NeurIPS, ICML, ICLR, UAI, AISTATS, AAAI, Siggraph, CVPR or TPAMI/JMLR. WHAT WE OFFER Build what actually matters Help shape an AI-native engineering company at a formative stage, tackling problems that genuinely matter for industry and society. This is work with real-world impact - and something you can be proud to stand behind. Learn alongside exceptional people Work with a high-caliber, collaborative team of engineers, scientists, and operators who care deeply about doing great work, and about helping each other get better. We come from diverse backgrounds, but we share a commitment to operating at the highest level and addressing some of the most complex challenges out there. If you’re ambitious, thoughtful, and driven by impact, you’ll feel at home. Influence over hierarchy We operate with a flat structure: good ideas win - wherever they come from. Questioning assumptions and challenging the status quo isn’t just welcomed, it’s expected. Sustainable pace, long-term ambition Building meaningful technology is a marathon, not a sprint. We believe in balancing focused, ambitious work with a life beyond it. Our hybrid model blends time together in our Shoreditch office with work-from-home days, giving you the flexibility to work sustainably while staying connected in person. And it doesn’t stop there … 🚀 Equity options - share meaningfully in the company you’re helping to build. 🏦 10% employer pension contribution - because investing in future matters. 🍽️ Free office lunches - to keep you energised and focused. 👶 Enhanced parental leave - 3 months full pay paternity and 6 months full pay maternity leave, to provide extra flexibility during the moments that matter most. 🍼 YellowNest nursery scheme - to help working parents manage childcare costs. ☀️ 25 days of Annual Leave (+ Public Holidays) - because taking time to rest matters. 🏥 Private medical insurance - 100% employee cover, giving you complete peace of mind. 💪 Wellhub Subscription - gain access to thousands of gyms, classes and wellness apps, supporting both physical and mental wellbeing. 👀 Eye tests - because good work depends on good health. 📈 Personal development - dedicated support for learning, development, and leveling up over time. 💛 Employee Assistance Programme (EAP) - confidential wellbeing support, available whenever you need it. 🚲 Bike2Work scheme and 🚆 Season ticket loan - to make getting to work easier and greener. 🚗 Octopus EV salary sacrifice - for a simpler, more sustainable way to drive electric. 🔎 Watch this space, we’re continuing to build this as we grow…   We value diversity and are committed to equal employment opportunity regardless of sex, race, religion, ethnicity, nationality, disability, age, sexual orientation or gender identity. We strongly encourage individuals from groups traditionally underrepresented in tech to apply. To help make a change, we sponsor bright women from disadvantaged backgrounds through their university degrees in science and mathematics.    We collect diversity and inclusion data solely for the purpose of monitoring the effectiveness of our equal opportunities policies and ensuring compliance with UK employment and equality legislation. This information is confidential, used only in aggregate form, and will not influence the outcome of your application.

PhysicsX

ABOUT US PhysicsX is a deep-tech company with roots in numerical physics and Formula One, dedicated to accelerating hardware innovation at the speed of software. We are building an AI-driven simulation software stack for engineering and manufacturing across advanced industries. By enabling high-fidelity, multi-physics simulation through AI inference across the entire engineering lifecycle, PhysicsX unlocks new levels of optimization and automation in design, manufacturing, and operations — empowering engineers to push the boundaries of possibility. Our customers include leading innovators in Aerospace & Defense, Materials, Energy, Semiconductors, and Automotive. NOTE: WE ARE CURRENTLY RECRUITING FOR MULTIPLE LEVELS AND POSITIONS, HOWEVER PLEASE ONLY APPLY FOR THE ROLE THAT BEST ALIGNS WITH YOUR SKILLSET AND CAREER GOALS. WHAT YOU WILL DO  * Own Research work-streams at a high-level to deliver outcomes. * Align priorities with problem stakeholders, internal and external. * Set the technical direction for the stream and apply judgement and taste to drive progress. * Plan roadmaps with clear milestones for key decisions and outcomes. * Organise and guide the more junior members of the team to effectively execute and deliver against this roadmap. * Communicate purpose and key outcomes to raise awareness across the company and create opportunities for use and deployment. * Contribute towards Research group strategy and culture. * Identify research areas that would be valuable to the company and champion their development, ordering wrt other research objectives. * Promote effective working patterns and proactively flag issues with team dynamics to foster a productive environment. * Nurture younger colleagues to grow their skillset and guide their professional development. * The below activities in particular. * Work closely with our machine learning engineers, simulation engineers, and customers to translate physics and engineering challenges into mathematical problem formulations. * Build models to predict the behaviour of physical systems using state-of-the-art machine learning and deep learning techniques. * Discuss the results and implications of your work with colleagues and customers, especially how these results can address real-world problems. * Collaborate with colleagues beyond the research team to translate your models into production-ready code. * Communicate your work to others internally and externally as called for in paper publication venues, industry workshops, customer conversations, etc. This will involve writing for academic and non-academic audiences.   WHAT YOU BRING TO THE TABLE * Ability to scope and effectively deliver projects. * Enthusiasm about using machine learning, especially deep learning and/or probabilistic methods, for science and engineering. * Strong problem-solving skills and the ability to analyse issues, identify causes, and recommend solutions quickly. * Excellent collaboration and communication skills — with teams and customers alike. * PhD in computer science, machine learning, applied statistics, mathematics, physics, engineering, or a related field, with particular expertise in any of the following: 1. operator learning (neural operators), or other probabilistic methods for PDEs; 2. geometric deep learning or other 3D computer vision methods for point-cloud or mesh-structured data; 3. generative models for geometry and spatiotemporal data (VAEs, Diffusion Models, Bayesian non-parametric, scaling to large datasets, etc.). * Ideally, >4 years of experience in a data-driven role in a professional industry setting, where you have been instrumental in: * building machine learning models and pipelines in Python, using common libraries and frameworks (PyTorch / CUDA, ideally with exposure to JAX, NumPy / SciPy), especially including deep learning applications; * developing models for bespoke problem settings that involve high-dimensional data (spatiotemporal, geometric, physical); * iterating on network architectures and model structure, tuning and optimising for inductive biases, improved generalisability, and improved performance; * combining theoretical reasoning with empirical intuition to guide investigation; * formulating and running experiment pipelines to benchmark models and produce comparable results; * writing skills for communication complex technical concepts to peers and non-peers, tailoring the message for the required audience. * Publication record in reputable venues that demonstrates mastery in your field, and in particular the domains of interest listed above. Desirable venues include (but not limited to): NeurIPS, ICML, ICLR, UAI, AISTATS, AAAI, Siggraph, CVPR or TPAMI/JMLR.   WHAT WE OFFER Build what actually matters Help shape an AI-native engineering company at a formative stage, tackling problems that genuinely matter for industry and society. This is work with real-world impact - and something you can be proud to stand behind. Learn alongside exceptional people Work with a high-caliber, collaborative team of engineers, scientists, and operators who care deeply about doing great work, and about helping each other get better. We come from diverse backgrounds, but we share a commitment to operating at the highest level and addressing some of the most complex challenges out there. If you’re ambitious, thoughtful, and driven by impact, you’ll feel at home. Influence over hierarchy We operate with a flat structure: good ideas win - wherever they come from. Questioning assumptions and challenging the status quo isn’t just welcomed, it’s expected. Sustainable pace, long-term ambition Building meaningful technology is a marathon, not a sprint. We believe in balancing focused, ambitious work with a life beyond it. Our hybrid model blends time together in our Shoreditch office with work-from-home days, giving you the flexibility to work sustainably while staying connected in person. And it doesn’t stop there … 🚀 Equity options - share meaningfully in the company you’re helping to build. 🏦 10% employer pension contribution - because investing in future matters. 🍽️ Free office lunches - to keep you energised and focused. 👶 Enhanced parental leave - 3 months full pay paternity and 6 months full pay maternity leave, to provide extra flexibility during the moments that matter most. 🍼 YellowNest nursery scheme - to help working parents manage childcare costs. ☀️ 25 days of Annual Leave (+ Public Holidays) - because taking time to rest matters. 🏥 Private medical insurance - 100% employee cover, giving you complete peace of mind. 💪 Wellhub Subscription - gain access to thousands of gyms, classes and wellness apps, supporting both physical and mental wellbeing. 👀 Eye tests - because good work depends on good health. 📈 Personal development - dedicated support for learning, development, and leveling up over time. 💛 Employee Assistance Programme (EAP) - confidential wellbeing support, available whenever you need it. 🚲 Bike2Work scheme and 🚆 Season ticket loan - to make getting to work easier and greener. 🚗 Octopus EV salary sacrifice - for a simpler, more sustainable way to drive electric. 🔎 Watch this space, we’re continuing to build this as we grow…   We value diversity and are committed to equal employment opportunity regardless of sex, race, religion, ethnicity, nationality, disability, age, sexual orientation or gender identity. We strongly encourage individuals from groups traditionally underrepresented in tech to apply. To help make a change, we sponsor bright women from disadvantaged backgrounds through their university degrees in science and mathematics.    We collect diversity and inclusion data solely for the purpose of monitoring the effectiveness of our equal opportunities policies and ensuring compliance with UK employment and equality legislation. This information is confidential, used only in aggregate form, and will not influence the outcome of your application.