Software Engineer, Data Infrastructure

About Us

Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation AI models and end-to-end products designed to solve real-world business problems. We train and deploy frontier models for enterprises building AI systems. Our work is instrumental to the widespread adoption of AI, and we're looking for people who want to be part of that mission.

We obsess over what we build. Each team member contributes to increasing the capabilities of our models and the value they drive for our customers. Cohere is a global team of researchers, engineers, designers, and more, passionate about their craft. Headquartered in Toronto, we have key offices in London, New York City, San Francisco, Montreal, Paris, Berlin, and Seoul.

About the Role

We're building the data infrastructure behind some of the most demanding AI training workloads in the world. In this role, you'll help construct and maintain the high-performance data layer our Modeling teams rely on for training and evaluation jobs.

Responsibilities

• Work directly on petabyte-scale storage infrastructure, addressing networking and performance challenges.
• Collaborate daily with researchers and engineers who are leaders in their fields.

Requirements

• 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.).
• 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 Skills

• Familiarity with modern analytics tooling such as BigQuery, Airflow, or dbt.
• Genuine excitement about AI—you follow the research, have opinions, and enjoy being in the weeds.
• Comfort operating at the edge of what's known, with a desire to build something genuinely new rather than optimize what already exists.

Benefits

• A weekly lunch stipend of $75/£75 or equivalent in your local currency.
• Full health and dental benefits, including a separate budget for mental health.
• RRSP matching, 401K, or Pension Scheme contributions.
• 100% Parental Leave top-up for up to 6 months for either parent.
• Annual enrichment benefits: Arts & culture, fitness/wellness, quality time, and a workspace improvement credit.
• Education & learning stipend for conferences, courses, and coaching.
• 6 weeks of paid vacation (30 working days).
• Budget for traveling to other offices if you are remote, plus an annual company offsite.
• $500 home office stipend to set up your workspace.

Schedule

This is a full-time position.

Work Arrangements

Cohere is remote-friendly with offices in Toronto, London, New York City, San Francisco, Montreal, Paris, Berlin, and Seoul.

• For those in the office: Daily lunch program, plenty of snacks, and regular community and social events.
• For those not near an office: Co-working benefit to work alongside others in your city.