Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Staff Software Engineer - AI Research Infrastructure

$190k - $270k

Databricks Inc.

Staff Software Engineer - AI Research InfrastructureP-1215At Databricks, we are obsessed with enabling data teams to solve the world’s toughest problems, from security threat detection to cancer drug development. We do this by building and running the world’s best data and AI platform so our customers can focus on the high-value challenges that are central to their own missions.The Databricks AI Research organization enables companies to develop AI models and agents using their own data, with technologies ranging from post-training open source LLMs to developing advanced multi-agent architectures. Databricks AI does so by producing novel science and putting it into production. Databricks AI is committed to the belief that a company’s AI models and agents are just as valuable as any other core IP, and that high-quality AI should be available to all.Job DescriptionAs a Staff Software Engineer, AI Research Infrastructure, you will be developing and running the research stack that powers Databricks AI Research. You will design and build services that schedule, orchestrate, and observe large‑scale training and inference experiment workloads across thousands of GPUs, improve our dev tooling and ensure that researchers can iterate quickly without sacrificing reliability, efficiency, or security.You’ll partner closely with research scientists, ML engineers, and platform teams to turn experimental workloads into robust, repeatable pipelines, and to push the limits of what our infrastructure can support.The Impact you will haveAs a Staff Software Engineer on the AI Research Infra Team at Databricks, you will: Design and implement infrastructure that supports large‑scale experiments, data processing, and model training (e.g., HPC clusters, GPU fleets, or cloud‑based systems)Enable researchers to go from idea to large‑scale experiment in minutes, not days, by building powerful abstractions for job submission, scheduling, and monitoring.Create tooling that improves research developer productivity, such as experiment management systems, CI/testing infrastructure for research code, and workflows that reduce iteration time.Influence the long‑term roadmap for research computation, shaping how Databricks AI Research train, evaluate, and ship models to customers.Serve as a technical mentor and force multiplier for other engineers working on compute, infra, and AI systems.What We Look forBS/MS or PhD in Computer Science or related field5+ years of software engineering experience, including substantial time working on large‑scale distributed systems or infrastructure.Have deep experience with building and operating distributed systems, data pipelines, or large‑scale backend services, ideally involving GPUs, clusters, or major cloud providers.Are proficient in one or more systems programming languages (e.g., C++, Rust, Go, Java, Scala) and can design, implement, and debug complex services.Have built or significantly contributed to cluster schedulers, resource managers, or large‑scale job orchestration systems (e.g., Kubernetes, Slurm, Ray, custom internal systems).Understand modern ML training and inference workflows (e.g., distributed training, model parallelism, fine‑tuning, evaluation), even if you’re not primarily a research scientist.Can move fast and be pragmatic in getting things done, while caring about operational excellence. Have driven complex systems from prototype to stable, well‑owned services.Communicate clearly with both researchers and engineers, and enjoy translating between research needs and infra realities.Pay Range TransparencyDatabricks is committed to fair and equitable compensation practices. The pay range(s) for this role is listed below and represents the expected salary range for non-commissionable roles or on-target earnings for commissionable roles. Actual compensation packages are based on several factors that are unique to each candidate, including but not limited to job-related skills, depth of experience, relevant certifications and training, and specific work location. Based on the factors above, Databricks anticipates utilizing the full width of the range. The total compensation package for this position may also include eligibility for annual performance bonus, equity, and the benefits listed above. For more information regarding which range your location is in visit our page here.Local Pay Range$190,000—$270,000 USDAbout DatabricksDatabricks is the Data and AI company. More than 20,000 organizations worldwide — including adidas, AT&T, Bayer, Block, Mastercard, Rivian, Unilever, and 70% of the Fortune 500 — rely on the Databricks Data + AI Platform to build and scale data and AI apps, analytics and agents. Headquartered in San Francisco with 30+ offices around the globe, Databricks offers a unified platform that includes Genie, Lakebase, Agent Bricks, Lakeflow, Lakehouse, and Unity Catalog. To learn more, follow Databricks on LinkedIn, X, YouTube, and Instagram.BenefitsAt Databricks, we strive to provide comprehensive benefits and perks that meet the needs of all of our employees. For specific details on the benefits offered in your region click here.Our Commitment to Diversity and InclusionAt Databricks, we are committed to fostering a diverse and inclusive culture where everyone can excel. We take great care to ensure that our hiring practices are inclusive and meet equal employment opportunity standards. Individuals looking for employment at Databricks are considered without regard to age, color, disability, ethnicity, family or marital status, gender identity or expression, language, national origin, physical and mental ability, political affiliation, race, religion, sexual orientation, socio-economic status, veteran status, and other protected characteristics.ComplianceIf access to export-controlled technology or source code is required for performance of job duties, it is within Employer's discretion whether to apply for a U.S. government license for such positions, and Employer may decline to proceed with an applicant on this basis alone.Applicant Privacy Notice

Vacancy posted 4 days ago
Similar jobs that could be interesting for youBased on the Staff Software Engineer - AI Research Infrastructure in New York, NY vacancy
  • $188k - $275k

     ...Essential Cloud for AI. Built for...  ...combines superior infrastructure performance with deep...  ...leading companies, research labs, and universities...  ...challenging engineering problems, and are...  ...About the Role: As a Staff Engineer on Marimo...  ...of experience in software engineeringStrong... 
    Suggested
    Permanent employment
    Full time
    Temporary work
    Casual work
    Work at office
    Worldwide
    Flexible hours

    CoreWeave

    New York, NY
    2 days ago
  •  ...security-first enterprise AI company. We build cutting-edge...  .... Cohere is a team of researchers, engineers, designers, and more, who are...  ...looking for Members of Technical Staff to join the Model Serving...  ...running production infrastructure at a large scaleExperience... 
    Suggested
    Full time
    Work experience placement
    Work at office
    Local area
    Remote work
    Home office

    Cohere

    New York, NY
    5 days ago
  • $216.2k - $270.25k

    Scale GP is building the infrastructure that makes enterprise AI seamless. We are looking for a Senior or Staff Infrastructure Engineer to act as a primary technical lead, engineering the...  ...role, with 5+ years of full-time software engineering experience.Deep understanding... 
    Suggested
    Full time

    Scale AI

    New York, NY
    5 days ago
  • $257k - $386k

     ...team in the creator economy and are looking for a Senior/Staff Software Engineer, Data Infrastructure to support our mission.This role is remote, with...  ...streaming or infrastructure tooling.Experience supporting ML/AI workflows, training pipelines, or evaluation systems.We... 
    Suggested
    Work at office
    Local area
    Remote work
    Worldwide
    Flexible hours
    2 days per week

    Patreon

    New York, NY
    3 days ago
  • Perplexity AI Inc. is seeking an AI-focused engineer for its Acceleration team. This role involves identifying high-potential AI opportunities and enhancing collaboration across teams in a hybrid work environment. The ideal candidate will have 5 to 15+ years of experience... 
    Suggested

    Perplexity AI Inc.

    New York, NY
    3 days ago
  • $405k

    Anthropic is seeking a Staff Software Engineer to lead technical direction within the Claude Code team...  ...role, you’ll partner closely with researchers and take full ownership of crucial systems...  ...architecture and a passion for AI research. Competitive salary range between... 
    Flexible hours

    Anthropic

    New York, NY
    1 day ago
  • $244k - $305k

    Role Summary:Datadog is seeking a Staff Software Engineer to help shape the future of our Bring Your...  ...customers deploy and manage in their own infrastructure. This role will focus on building and...  ...and security platform for the AI era, providing businesses with unified... 
    Work at office

    Datadog

    New York, NY
    2 days ago
  • $187k - $234k

     ...Addepar is a global data and AI platform empowering investment...  ...integrates with nearly 650 software, data and consulting partners...  ...experienced Backend Software Engineer with a strong Java background...  ...cloud context, with automated infrastructure and service-oriented architecture... 
    Full time
    H1b
    Remote work
    Worldwide
    Visa sponsorship
    Work visa

    Addepar

    New York, NY
    2 days ago
  • $234k - $300k

     ...monitor, explain, and improve AI systems in production,...  ...ship AI with confidence.As a Staff Engineer, you’ll lead the development...  ...understanding of both AI systems and software engineering to solve open-...  .... It brings applications, infrastructure, data, models, and security... 
    Work at office

    Datadog

    New York, NY
    5 days ago
  •  ...StripeStripe is a financial infrastructure platform for businesses....  ...product teams building AI-first solutions to some of...  .... We’re a mix of machine engineering engineers and full-stack software engineers impacting ~...  ...needs.What you'll doAs a Staff Engineer on Payment Intelligence... 

    Stripe

    New York, NY
    1 day ago
  • $172.2k - $224.3k

     ...validates, and commercializes software as prescription medical...  ...clinicians, scientists, researchers, designers, technologists, engineers and more, united in a...  ..., you are a pioneer in AI-augmented engineering;...  ...mobile app, backend, and infrastructure boundaries. Contribute... 
    Permanent employment
    Full time
    Temporary work
    Work at office
    Local area
    Visa sponsorship
    Flexible hours

    Click Therapeutics

    New York, NY
    2 days ago
  • $207k - $301k

     ...Develop scalable automated rating infrastructure, high-quality evaluation...  ...experience.8 years of experience in software development.Experience integrating generative AI tools or LLM interfaces into...  ...collaborating across engineering teams to prioritize and execute... 
    Shift work
    Early shift

    Google

    New York, NY
    4 days ago
  • $205k - $275k

     ...Attentive® is the AI marketing platform for 1:1 personalization...  ...AI-powered personalization engine delivers bespoke experiences...  .... We're looking for a Staff Software Engineer to serve as a...  ...services, APIs, and shared infrastructure that multiple teams depend on... 
    Full time
    Work at office

    Attentive

    New York, NY
    3 days ago
  • $211.1k - $285k

     ...asset is our people. As a Staff Software Engineer on the Billing team, you’ll...  ...our core marketplace to new AI-driven innovations like Zo,...  ...projects from the early stages of research and technical design...  ...desire to own your team’s infrastructure (as code!).The mentality of... 
    Remote work
    Flexible hours

    ZocDoc

    New York, NY
    5 days ago
  • $194.06k - $258.75k

     ...ecosystem of devices and cloud software. Like our products, we work...  ...that combines modern 911 infrastructure with an AI intelligence layer—helping...  ...the actual call-handling engine, responsible for every packet...  ...life-critical system.As a Staff UC Software Engineer, you will... 
    Work experience placement
    Work at office

    Axon

    New York, NY
    2 days ago
  • $188k - $275k

     ...CoreWeave is The Essential Cloud for AI™. Built for pioneers by pioneers, CoreWeave...  ..., CoreWeave combines superior infrastructure performance with deep technical expertise...  ...Learn more at What you’ll do As a Staff Software Engineer on the Identity & Access Management (... 
    Permanent employment
    Full time
    Temporary work
    Casual work
    Work at office
    Immediate start
    Flexible hours

    CoreWeave

    New York, NY
    2 days ago
  • $195k - $408k

     ...team operates with the high engineering rigor that this demands....  ...RoleWe're hiring a Senior Software Engineer (Staff scope) to help scale and evolve...  ...time off.How will I use AI at Roku? At Roku, we don’t...  ...operating in cloud infrastructure (GCP or AWS) at scaleExposure... 
    Work at office
    Local area
    Immediate start
    Remote work
    Monday to Thursday
    Flexible hours

    Roku

    New York, NY
    4 days ago
  • $240k - $300k

    Traba is the AI operating layer for the industrial supply chain. We started in workforce—temp staffing...  ...it real across the stack.) This is a platform-engineering role first, so you'll go where the priorities are: the infrastructure an agent needs one quarter, a matching, data,... 
    Temporary work
    Local area
    Shift work

    Traba

    New York, NY
    1 day ago
  • $228k - $240k

     ...seller and buyer clients. As a Staff Software Engineer in the Transaction Journey (TJ) organization...  ...technical debt, manual toil, and infrastructure costs. Adopt and operationalize...  ...software patterns, including integrating AI, data pipelines, and telemetry systems... 
    Minimum wage
    Full time
    Flexible hours

    Compass

    New York, NY
    3 days ago
  • $240k - $285k

     ...bill pay, and travel software, Brex enables founders...  ...effortlessly. Brex’s AI-native automation and...  ...need to grow your career.Engineering at BrexEngineering at...  ...leaders.What you’ll doAs a Staff Software Engineer in...  ...and core financial infrastructure, driving architecture,... 
    Bank staff
    Work at office
    Remote work
    Work from home
    3 days per week

    Brex

    New York, NY
    2 days ago
  • $207k - $300k

     ...co-design future processors.Research high-level representations...  ...experience testing, and launching software products.5 years of...  ...TPUs/GPUs).Google's software engineers develop the next-generation...  ...(SW) co-design.The AI and Infrastructure team is redefining what’s possible... 
    Worldwide

    Google

    New York, NY
    7 days ago
  • $190.9k - $253.75k

     ...improve their business. Founded by engineers — and customer obsessed — we...  ...to scaling our services and infrastructure across millions of virtual...  ...of Databricks in data & AI to deliver vertical AI applications...  ...-on front-end and full-stack software development while working... 
    Work at office
    Local area
    Worldwide

    DataBricks

    New York, NY
    5 days ago
  • $188k - $275k

     ...The Essential Cloud for AI™. Built for pioneers by...  ...combines superior infrastructure performance with deep technical...  ...'s leading AI labs, researchers, and enterprises depend...  ...that enable every engineer at CoreWeave to build and ship software faster, more safely, and... 
    Permanent employment
    Full time
    Temporary work
    Casual work
    Work at office
    Remote work
    Flexible hours

    CoreWeave

    New York, NY
    2 days ago
  • $207k - $275k

     ...is The Essential Cloud for AI. Built for pioneers by pioneers...  ...CoreWeave combines superior infrastructure performance with deep...  ...The Platform & Infrastructure Engineering team sits at the core of our...  ...depends on.About the role:As a Staff Software Engineer, you will take end-... 
    Permanent employment
    Full time
    Temporary work
    Casual work
    Work at office
    Flexible hours

    CoreWeave

    New York, NY
    3 days ago
  • $50 per hour

    Staff Software EngineerThe global capital markets are among the largest...  ...legal terms.As a Staff Software Engineer, Full Stack at Noetica, you'...  ...with a talented team of AI/ML PhDs, legal SMEs, and market...  ...of Noetica's core infrastructure.About the RoleIn this opportunity... 
    Full time
    Work at office
    Local area
    Flexible hours
    2 days per week
    3 days per week

    Thomson Reuters

    New York, NY
    3 days ago
  • $192k - $252k

    Hi, we're Oscar. We're hiring Staff Software Engineers to join our Engineering team.Oscar is the first...  ...and maintainable applications, infrastructure, and interfaces that make interacting...  ...reimbursements.Artificial Intelligence (AI): Our AI Guidelines outline the acceptable... 
    Full time
    Work at office
    Flexible hours

    Oscar Health Insurance

    New York, NY
    5 days ago
  • $197.3k - $313.7k

     ...SalesforceSalesforce is the #1 AI CRM, where humans with...  ..., and large-scale engineering. The systems we build...  ...'re looking for a Staff Android Engineer to help...  ..., Design, Backend, Infrastructure, iOS, and fellow mobile...  ...years of professional software engineering experience... 
    Full time
    Immediate start
    Worldwide

    Salesforce

    New York, NY
    4 days ago
  • $107k - $168k

     ...Addepar is a global data and AI platform empowering...  ...integrates with nearly 650 software, data and consulting partners...  ...We are currently seeking a Staff Software Engineer to join the Calculation Engines...  ...traffic flows through this infrastructure; you will be responsible for... 
    Full time
    Remote work
    Worldwide

    Addepar

    New York, NY
    2 days ago
  • $207k - $275k

    CoreWeave is The Essential Cloud for AI. Built for pioneers by pioneers, CoreWeave...  ..., CoreWeave combines superior infrastructure performance with deep technical expertise...  ...25. Learn more at .About the RoleAs a Staff Software Engineer within our Compute Architecture organization... 
    Permanent employment
    Full time
    Temporary work
    Casual work
    Work at office
    Flexible hours

    CoreWeave

    New York, NY
    4 days ago
  • $244k - $305k

     ...platform in the world for engineers to understand and...  ...Opportunity:Datadog’s Staff Engineers are our technical...  ...wide standards around software and systems...  ...demonstrated ability to use AI coding tools in day-to...  ...brings applications, infrastructure, data, models, and security... 

    Datadog

    New York, NY
    5 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Staff Software Engineer - AI Research Infrastructure. Be the first to apply!