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.

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.

Own the product roadmap for AI workload and agent runtime orchestration across Kubernetes clusters and heterogeneous accelerators. Responsibilities include deployment APIs, workload placement, capacity, autoscaling, isolation, networking, governance, metering, reliability, and auditability. The role spans two engineering pods and requires deep expertise in Kubernetes, GPUs, distributed systems, multi-tenancy, platform APIs, and technical prototyping, with influence across matrixed teams and no direct reports.