AI Researcher - Full time
Job description
AI Researcher at an AI-native real estate company, leading research on a foundation diffusion model for floorplan generation. The role owns research directions end-to-end, from discrete/continuous diffusion and flow matching to deployment in a product used by real estate professionals.
At a glance
- Employment
- Full-time
- Work mode
- On-site · Paris, France
What you'll do
- Design the core model architecture and work on a discrete design space for architectural layouts, choosing how to represent floorplans (e.g. as graphs of rooms/connections or token grids) so diffusion outputs are editable and code-compliant by construction
- Ensure the model can enforce architectural rules (e.g. room sizes, adjacency constraints) within its generation process
- Lead the training of a foundation diffusion model from scratch on GPU clusters, setting up distributed training across multiple nodes, optimizing data loading and checkpointing, and managing experiments at scale
- Evaluate the model's performance and establish benchmarks to measure success, then iterate on the model to push performance beyond existing methods
- Collaborate daily with architects to turn foundational research into deployable tools
What we're looking for
- PhD or Master's in Maths, Computer Science, Machine Learning, or a related field, or equivalent experience
- Strong foundations in ML and a track record of innovative research, publications, or high-impact projects
- Deep understanding of diffusion models (discrete or continuous), guided generation techniques, and the latest advances in generative modeling
- Proven experience training large-scale deep learning models on GPU clusters, with distributed training, multi-node jobs, experiment management, and handling large datasets
- Proficiency in Python with experience in PyTorch or similar frameworks
- Ability to write efficient, maintainable code and optimize training pipelines
- Ability to read, evaluate, and implement advanced ML research and adapt cutting-edge methods to domain-specific problems
- Strong teamwork skills and ability to articulate complex concepts clearly and work closely with AI engineers, researchers, and domain experts
Nice to have
- Background in a top AI research organization or cutting-edge startup, especially working on foundation models or generative AI tools
- Research publications in generative modeling (diffusion, VAEs, flows, GANs) or significant open-source contributions
- Interest in architecture or design
- Experience with RL, reward modeling, or constrained optimization (particularly MCTS, GRPO and RLHF)
- Experience with graph-based, sequence-based, or discrete structured generative models (e.g., molecule generation, layout generation, program synthesis) or with graph neural networks
- Familiarity with distributed training libraries (e.g., PyTorch Lightning)
Apply
Direct apply at Meetdavis
Opens the job posting at Meetdavis
About Davis
AI-native real estate company accelerating early-stage development and architectural design
Davis is an AI-native real estate company accelerating early-stage development and architectural design. It turns every input that shapes a development decision into decision-ready outputs: investor-grade feasibility studies, investment analysis, and architect-certified designs, delivered in days. Every stage pairs proprietary AI systems with expert review.
Mission
Build a foundation model for architectural design that generates compliant, editable building layouts from scratch
Key facts
- Real estate / AI
- Closed a $5.5M pre-seed co-led by Heartcore Capital and Balderton Capital, with Yellow, Evantic and Entrepreneur First, alongside angels from the founding teams of Spacemaker, Black Forest Labs, Huggi
- Works with leading developers and expects to support hundreds of projects over the coming year
- Team of 12 with technical profiles from Polytechnique, ENS and INRIA alongside architects
- Started working together in July 2025 at Entrepreneur First's first European residency
Partners & customers