Software Engineer, Data Infrastructure

Cohere

Software Engineer building data infrastructure for AI training workloads. Collaborating with top researchers and engineers in cutting-edge AI.

Posted 5/15/2026full-timeRemote • New York • 🇺🇸 United StatesMid-LevelSeniorWebsite

Tech Stack

Tools & technologies

AirflowApacheBigQueryKubernetesPythonSpark

About the role

Key responsibilities & impact

Build and maintain the high-performance data layer our Modeling teams rely on for training and evaluation jobs.
Work directly on petabyte-scale storage infrastructure, and the networking and performance challenges that come with it.
Collaborate daily with researchers and engineers who are some of the best in the world at what they do.

Requirements

What you’ll need

4+ years of experience working on data storage infrastructure
Strong command of Python
Kubernetes experience, especially on the storage side (Persistent Volumes, CSI drivers, etc.)
The ability to transform unstructured data into performant datasets across diverse storage backends including S3, GCS, and POSIX
Experience with distributed data processing frameworks such as Apache Beam, Spark, or Flink
[Nice-to-have] Familiarity with modern analytics tooling such as BigQuery, Airflow, or dbt
Genuine excitement about AI.
Comfort operating at the edge of what's known, with a desire to build something genuinely new rather than optimize what already exists

Benefits

Comp & perks

An open and inclusive culture and work environment
Work closely with a team on the cutting edge of AI research
Weekly lunch stipend, in-office lunches & snacks
Full health and dental benefits, including a separate budget to take care of your mental health
100% Parental Leave top-up for up to 6 months
Personal enrichment benefits towards arts and culture, fitness and well-being, quality time, and workspace improvement
Remote-flexible, offices in Toronto, New York, San Francisco, London and Paris, as well as a co-working stipend
6 weeks of vacation (30 working days!)

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

PythonKubernetesApache BeamSparkFlinkdata storage infrastructuredata processing frameworksunstructured data transformationperformance optimizationdata layer development

Soft Skills

collaborationproblem-solvinginnovationadaptabilitycommunicationteamworkcreativityenthusiasm for AIdesire to buildoperating at the edge of knowledge