Teams at DataRobot

Conduct research engineering for a probabilistic foundation model spanning temporal, tabular, and mixed-modality data. Design model architectures, encoders, attention mechanisms, distributional output heads, and decoding strategies; implement and train scalable PyTorch models; run controlled experiments, ablations, and performance profiling; diagnose training behavior; and produce reproducible, production-quality code. Candidates with stochastic modeling expertise may also develop synthetic data generators and simulate richer stochastic dynamics.

Lead the delivery of advanced customer initiatives for AI solutions, manage project risks, and ensure successful client engagements involving predictive and generative AI.

As a Staff Software Engineer, you will lead the technical vision for DataRobot's Fleet team, mentoring engineers and architecting scalable infrastructure using Kubernetes for multi-cloud environments.