Software Engineer, Data Infrastructure

Cohere is on a mission to scale intelligence to serve humanity by deploying frontier models for AI systems. The Software Engineer, Data Infrastructure will build and maintain high-performance data layers essential for training and evaluation jobs, working on petabyte-scale storage infrastructure and collaborating with top researchers and engineers in the field.

Responsibilities

• 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

Preferred

• Familiarity with modern analytics tooling such as BigQuery, Airflow, or dbt
• BigQuery
• Airflow
• dbt

Benefits

• 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!)

Company

• Secure enterprise AI models and software for private business deployment
• Toronto, Ontario, CAN
• https://cohere.com/
• Founded in 2019
• 501-1000 employees

Qualifications

• 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
• 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
• Familiarity with modern analytics tooling such as BigQuery, Airflow, or dbt

Required skills

• Python, Kubernetes, Persistent Volumes, CSI drivers, S3, GCS, POSIX, Apache Beam, Spark, Flink

Experience

• 4+ years