Software Engineer, ML Production
Toronto
Cohere
Cohere provides industry-leading large language models (LLMs) and RAG capabilities, tailored to meet the needs of enterprise use cases that solve real-world problems.We recently raised our Series B, signed a multi-year partnership with Google Cloud, and we are focused on bringing our technology to market. We will partner with customers so they can build natural language understanding and generation into their products with just a few lines of code.
We’re ambitious — we believe our technology will fundamentally transform how industries interact with natural language. And we have the technical chops to back it up - Cohere’s CEO, Aidan Gomez, is a co-author of the groundbreaking paper “Attention is all you need”, (over 23k citations) and was previously part of Google Brain. Our entire technical team is world-class.
We are focused on creating a diverse and inclusive work environment so that all of our team members can thrive. We welcome kind and brilliant people to our team, from wherever they come.
Why this role?This role sits at the intersection between machine learning and software engineering. It is a high-impact position in delivering large-scale language models that Cohere builds. This role will focus on system designs of model inference runtimes and model production pipelines to ensure that our models run efficiently and reliably in various environments. In addition, this role will work closely with the latest ML frameworks to integrate cutting-edge model runtimes.
Please Note: We have offices in Toronto, Palo Alto, and London but embrace being remote-first! We are looking for candidates based in Canada for this role.
As a Software Engineer, you will:
- Build model inference framework and infrastructure.
- Build model production pipeline tools.
- Integrate various third party model runtimes.
- Build model performance profiling tools.
- Optimize frameworks and improve system design.
You may be a good fit if you have:
- Extensive software engineering experience writing production-quality code.
- Experience optimizing systems and tools.
- Basic understanding of machine learning workflow.
- Bonus: experience in at least one ML framework such as Tensorflow, PyTorch, or JAXBonus: experience writing kernels for GPUs using CUDA.
We welcome applicants of all kinds and are committed to providing both an equal opportunity process and work environment. We value and celebrate diversity and strive to create an inclusive work environment for all.
Our Perks: 🤝 An open and inclusive culture and work environment 🧑💻 Work closely with a team on the cutting edge of AI research 🍽 Free daily lunch 🦷 Full health and dental benefits for employees based in Canada and the US, including a separate budget to take care of your mental health 🐣 100% Parental Leave top-up for 6 months for employees based in Canada and 12 weeks in the US🎨 Personal enrichment benefits towards arts and culture, fitness and well-being, quality time, and workspace improvement🏙 Remote-flexible, offices in Toronto and Palo Alto and coworking stipend✈️ 6 weeks of vacation and shared Canada/US holidays
#LI-Remote
Tags: APIs CUDA Engineering GCP Google Cloud Machine Learning Model inference Pipelines PyTorch Research TensorFlow
Perks/benefits: Career development Flex hours Flex vacation Health care Home office stipend Parental leave Salary bonus
More jobs like this
Explore more AI, ML, Data Science career opportunities
Find even more open roles in Artificial Intelligence (AI), Machine Learning (ML), Natural Language Processing (NLP), Computer Vision (CV), Data Engineering, Data Analytics, Big Data, and Data Science in general - ordered by popularity of job title or skills, toolset and products used - below.
- Open Lead Data Analyst jobs
- Open MLOps Engineer jobs
- Open Data Science Manager jobs
- Open Senior Business Intelligence Analyst jobs
- Open Data Manager jobs
- Open Data Engineer II jobs
- Open Power BI Developer jobs
- Open Sr Data Engineer jobs
- Open Principal Data Engineer jobs
- Open Data Analytics Engineer jobs
- Open Junior Data Scientist jobs
- Open Business Intelligence Developer jobs
- Open Data Scientist II jobs
- Open Senior Data Architect jobs
- Open Product Data Analyst jobs
- Open Business Data Analyst jobs
- Open Sr. Data Scientist jobs
- Open Data Analyst Intern jobs
- Open Big Data Engineer jobs
- Open Manager, Data Engineering jobs
- Open Junior Data Engineer jobs
- Open Azure Data Engineer jobs
- Open Data Quality Analyst jobs
- Open Data Product Manager jobs
- Open ETL Developer jobs
- Open Data quality-related jobs
- Open Business Intelligence-related jobs
- Open ML models-related jobs
- Open Data management-related jobs
- Open GCP-related jobs
- Open Java-related jobs
- Open Privacy-related jobs
- Open Finance-related jobs
- Open Data visualization-related jobs
- Open APIs-related jobs
- Open Deep Learning-related jobs
- Open PyTorch-related jobs
- Open Snowflake-related jobs
- Open Consulting-related jobs
- Open TensorFlow-related jobs
- Open PhD-related jobs
- Open CI/CD-related jobs
- Open NLP-related jobs
- Open Data governance-related jobs
- Open Kubernetes-related jobs
- Open Airflow-related jobs
- Open LLMs-related jobs
- Open Data warehouse-related jobs
- Open Databricks-related jobs
- Open Hadoop-related jobs