AI Research Scientist / Engineer – Generative World Models

The role

You will drive research and development for Cassandra, a Generative World Model for healthcare. You will work on pre-training and scaling the model and developing downstream applications like fall prediction.

What you will do

• Improve the end-to-end pre-training pipeline including data representation, tokenization, architecture, and scaling.
• Design experiments, develop new approaches, and translate research into model improvements.
• Improve predictive capabilities for resident fall risk.
• Explore new applications such as forecasting resident behavior changes and simulating care interventions.
• Develop and productionize downstream applications.

Required qualifications

• Hands-on experience training large-scale sequence models, such as Transformers, state-space models, or LLMs.
• Deep understanding of neural network architectures, optimization, training dynamics, and scaling laws.
• Strong Python and PyTorch skills.
• Experience building efficient, reliable training and evaluation pipelines.
• Ability to independently design experiments, interpret results, and translate research ideas into working implementations.
• Solid mathematical foundations in machine learning, probability, and statistics.
• Based in Copenhagen, Denmark, or willing to relocate.

Nice to have

• Research or engineering experience with LLM pre-training, including architecture design, data mixtures, training objectives, or scaling experiments.
• Experience with generative modeling of sequential, temporal, or event-based data.
• Familiarity with continuous-time models, point processes, time-to-event prediction, or survival modeling.
• Experience with long-context architectures, such as Mamba, recurrent models, or efficient attention mechanisms.
• Experience with distributed training, GPU optimization, and performance profiling.
• A track record of impactful ML research, demonstrated through publications, open-source contributions, or production systems.

Benefits

• Competitive Salary
• Participation in our warrant program
• Work with state-of-the-art technology in a pioneering field
• Large in-house training cluster
• In-office work place
• A vibrant, learning-focused work environment

About the company

Teton is building the foundational data layer for the point of care, using real-time, multimodal AI to generate a digital twin of the patient and care environment. The company uses a proprietary computer-vision system to automate documentation and manage resident acuity.