AI Researcher, World Models

Geolava; United States; Remote; USD 180,000 - 250,000; Full time

Benefits offered by this job

Medical, dental, vision insurance

Retirement plan

Global remote workforce

Job summary

Geolava is building spatial intelligence systems and is seeking an AI Researcher to own end-to-end research from framing questions to production-ready capabilities. You will develop world models using multi-temporal data and train large vision models to forecast asset changes over time.

The role emphasizes autonomous work on ambiguous problems in a startup context, with collaboration to ship capabilities for investors, lenders, and infrastructure owners.

Qualifications

• World-model research experience with academic or industry work.
• PhD in machine learning, computer vision, or closely related field.
• Fluency with deep learning for large vision models; diagnosing training runs.
• Empirical rigor with proper baselines and data splits.
• Ability to work autonomously on ambiguous problems in a small team.
• Clear written and spoken communication for technical and non-technical audiences.

Responsibilities

• Design, train, and evaluate world models of assets changing over time.
• Develop representation learning over multi-temporal imagery and data.
• Build rigorous evaluation with baselines, temporal splits, calibration, and leakage checks.
• Run experiments on cloud GPUs with attention to throughput and cost, produce writeups.
• Collaborate with AI and engineering teams to ship research as platform capabilities.
• Stay current with world-model, self-supervised, and generative modeling literature.

Skills

World-model research experience

Deep learning fluency

Empirical rigor

Ownership

Clear communication

Education

PhD in ML/CV or related field

Tools

PyTorch

Vision transformers

Diffusion models

Python

Cloud GPUs

Job description

At Geolava, we bring AI to the physical world. We are building spatial intelligence systems by leveraging world models to simulate the evolution of the physical world over time. We continuously model real-world dynamics, simulate actions, and forecast their impact on assets.

We are backed by top-tier VCs and have been revenue-generating from day one. This is your opportunity to join as a founding team member to help define the future of spatial intelligence from the ground up.

We are looking for an AI Researcher to help build Geolava's world model for the built world: a system that perceives the current state of properties and infrastructure, learns how they change over time, and forecasts what comes next, whether the driver is time, a renovation, a new construction, or a hurricane. You will work from raw earth observation and property data all the way to a model that can reason about state, action, and future state, and you will decide how we measure whether it works.

This is a research role on a small team, and you will own real problems end to end: framing the question, designing the experiment, running it on real data at scale, and writing up what you learned. It is also a startup. Research that works here does not sit in a paper. It ships as a capability inside the product that property investors, lenders, and infrastructure owners use to make decisions.

Responsibilities (What You'll Do)

• Design, train, and evaluate world models of the built environment that predict how assets change over time and in response to interventions and external forces.

• Develop representation learning and predictive modeling approaches over multi-temporal aerial and satellite imagery, property records, and other physical and administrative signals.

• Build rigorous evaluation for a domain with no public benchmarks: baselines, held-out temporal splits, calibration, and tests that distinguish genuine forecasting skill from leakage.

• Run experiments on cloud GPUs with attention to throughput and cost, and turn results into clear, decision-ready writeups.

• Work with the AI and engineering teams to take successful research into production as capabilities in Geolava's platform.

• Stay current with the world-model, self-supervised learning, and generative modeling literature, and bring the ideas that matter into our work.

Your Background

• World-model research experience. You have built or contributed to world models: learned dynamics models, latent predictive models, action-conditioned generation, or closely related work, in academia or industry. This is required.

• PhD. You hold a PhD in machine learning, computer vision, or a closely related field.

• Deep learning fluency. You are comfortable training large vision models, adapting foundation models efficiently, and diagnosing why a training run behaves the way it does.

• Empirical rigor. You design experiments to answer questions, insist on proper baselines and splits, and are honest about what the evidence does and does not show.

• Ownership. You are productive with ambiguous problems and small teams, and you drive work from idea through result without waiting for a roadmap.

• Clear communication. You write and speak precisely about technical work for both research and non-research audiences.

Technical Requirements

• PhD in machine learning, computer vision, or a related quantitative field.

• Demonstrated research work on world models or learned dynamics models, with publications, open-source work, or shipped systems to show for it.

• Expert-level PyTorch and experience training and fine-tuning vision transformers or diffusion models at scale.

• Strong Python and data engineering skills for large image and tabular datasets.

• Experience running training workloads on cloud GPU infrastructure.

• Experience with remote sensing, geospatial data, or spatiotemporal modeling is a strong plus.

Why Join Geolava?

• Be at the cutting edge of Physical AI.

• Competitive salary. Compensation varies according to candidate level and experience.

• Excellent benefits including medical, dental, and vision insurance, a retirement plan, short & long term disability and life insurance.

Geolava is committed to maintaining a drug-free workplace and promoting a safe, healthy working environment for all employees.

Geolava is an equal opportunity employer and has a global remote workforce. Applicants are considered solely based on their qualifications, without regard to an applicant's disability or need for accommodation. Any Geolava applicant who requires reasonable accommodations during the application process should contact Geolava's Human Resources Department to make the need for an accommodation known.