[Remote] Deep Learning Research Engineer Intern

Note: The job is a remote job and is open to candidates in USA. DataRobot is hiring a Deep Learning Research Engineer Intern to shape the architecture of a probabilistic foundation model for temporal, tabular, and mixed-modality tasks. The role focuses on designing, implementing, training, and evaluating PyTorch models, with opportunities to extend synthetic-data generation using stochastic dynamics.


Responsibilities

  • Designing and improving the core model architecture: input encoders for temporal, unordered tabular, and mixed-modality data; attention factorizations across variables, rows, and horizons; the distributional output heads and decoding strategies that produce coherent joint samples
  • Running controlled architecture studies, from ablations and scaling behavior to memory and throughput profiles, and turning the results into design decisions
  • Building scalable PyTorch implementations that support larger input and output spaces, better throughput, and tighter memory budgets
  • Studying how architectural choices, data-generation choices, and inference constraints interact to change benchmark quality and real-world usefulness
  • For candidates with the stochastic-modeling background: extending the synthetic-data engine with richer stochastic dynamics, constraints, dependence structures, heavy tails, and regime behavior
  • Turning research ideas into robust implementations and credible empirical results

Skills

  • Strong PyTorch skills and hands-on experience building and training deep models
  • Solid understanding of Transformers, attention variants, and long-context or state-space sequence architectures, including the tradeoffs between quality, memory, and latency
  • Experience with probabilistic modeling in neural networks: distributional output heads, likelihood-based or proper-scoring-rule losses, mixture or flow models, or related uncertainty-aware learning setups
  • Strong foundation in probability and statistics, enough to reason about densities and masses, dependence between variables, and calibration
  • Ability to connect architectural ideas to working GPU-native implementations, controlled experiments, and diagnostics that show why a change helped
  • Strong engineering habits: readable code, tests, reproducible experiments, and disciplined evaluation of model changes
  • Ability to debug training instability, reason about why results changed, and iterate quickly from hypothesis to evidence
  • Working knowledge of stochastic processes and stochastic differential equations: drift and diffusion, jumps, regime switching, mean reversion, heavy tails, and how such dynamics are simulated numerically
  • Experience designing synthetic data generators or simulation-based training curricula, and an understanding of how the training distribution shapes what a model learns
  • Depth in a domain with rich stochastic structure such as finance, energy, commodities, or a similarly quantitative field
  • Familiarity with tabular or mixed-modality deep learning: categorical, ordinal, count, and bounded targets alongside continuous ones

Benefits

  • Medical, Dental & Vision Insurance
  • Flexible Time Off Program
  • Paid Holidays
  • Paid Parental Leave
  • Global Employee Assistance Program (EAP)

Company Overview

  • DataRobot provides AI technology and ROI enablement services to global enterprises. It was founded in 2012, and is headquartered in Boston, Massachusetts, USA, with a workforce of 501-1000 employees. Its website is

  • Company H1B Sponsorship

  • DataRobot has a track record of offering H1B sponsorships, with 6 in 2026, 10 in 2025, 5 in 2024, 28 in 2023, 19 in 2022, 17 in 2021, 8 in 2020. Please note that this does not guarantee sponsorship for this specific role.

  • Back to blog

    Other Jobs To Apply

    Leave Administrator $17.97-$22.21 - Remote - Veterans Military Spouses

    Inpatient Coder - Community Hospital at Savista: 100% Remote (US)

    College Advisor (Part-Time, Contract - Remote)

    English Teacher at Pearson: 100% Remote (WA)

    Travel Support Assistance

    Claims Specialist Bodily Injury at National Interstate Insurance: 100% Remote (OH, MO)

    Bookkeeper for Remote Accounting Firm

    Remote Amazon Data Entry Jobs - No Experience - Part-Time

    Artificial Intelligence Solutions Director - Fargo Virtual Assistant for Business Banking

    Tesla Work From Home Jobs (Customer Service, Entry Level)

    Remote Travel Consultant

    Remote Fiction Editor — Flexible Schedule & Per-Project Pay

    Remote Clinical Pharmacist – Prior Authorization

    Remote Medical Scribe

    Remote Director of Drive-Thru Hardware Product Management

    Entry Level Review Writer Easy Phone Job - No Experience Required

    Product Manager IV, Laboratory (Remote)

    Service Desk Technician Level I - Remote

    Principal Java Engineer, Multiple Teams

    Remote Entry-Level Customer Support – No Experience Required

    Data Entry Computer Job - Work from Home Part Time

    Amazon Customer Service - Work From Home $16-$24/hr

    No-Code Automation Specialist (Zapier/Make.com)

    Senior Expert, ARM Kernel Developer

    Customer Service Rep (100% Work From Home)

    Permanent Job UPS Remote Jobs (Data Entry Specialist) – Full Time Job ID- 853 Job ID- 3076

    GTM Innovation Account Executive

    Virtual Care Veterinarian (Remote)

    Remote Work From Home Customer Service- Typing

    Research Associate (Remote)

    Remote Entry-Level Customer Support – No Experience Required

    Remote Library Assistant

    Technical Support Specialist (Remote) - Entry Level, No Degree Required, 15 - 18

    Remote Property Claims Adjuster

    Pinterest Growth Writer Remote Data Insights

    Clinical Neurologist (Remote | $200/hr)

    Associate Billing Representative - National Remote

    Remote | Senior NDT Inspector (Level II/III) — $70–$110/hour

    Student Employment Program Manager

    Remote Manager, Customer Success Engineering

    Senior Web Developer – Engineer

    Remote Customer Solutions Representative

    REMOTE Weekend OnCall Admin NEEDED For Home Care Agency In Irvine, CA

    Remote Customer Support Specialist

    RN Call Center Nurse – CA Remote

    Logistics Coordinator (Work From Home) - $1,800 to $3,500 WeeklyNorth Las Vegas

    Director, Internal Communications

    Work From Home

    Remote General Education Professor

    Eddie Bauer Work From Home Jobs (Customer Support) $75000/Year