Senior MLSys Engineer
Remote
OctoML
Build and scale production applications on the latest optimized models and fine tunes using the OctoAI SaaS or in your environment.Our platform, OctoAI, delivers generative AI infrastructure to run, tune, and scale models that power AI applications. OctoAI makes models work for you by providing developers easy access to efficient AI infrastructure so they can run the models they choose, tune them for their specific use case, and scale from dev to production seamlessly. With the fastest foundation models on the market (including Llama-2, Stable Diffusion, and SDXL), integrated customization solutions, and world-class ML systems under the hood, developers can focus on building apps that wow their customers without becoming AI infrastructure experts.
Our team consists of experts in cloud services, infrastructure, machine learning systems, hardware, and compilers as well as an accomplished go-to-market team with diverse backgrounds. We have secured over $130M in venture capital funding and will continue to grow over the next year. We're based largely in Seattle but have a remote-first culture with people working all over the US and elsewhere in the world.
We dream big but execute with focus and believe in creativity, productivity, and a balanced life. We value diversity in all dimensions and are always looking for talented people to join our team!
Overview: Our ML engineering team is dedicated to developing the most efficient and feature-packed engines for generative model deployment. This includes feature enablement and optimization for popular media models, such as Mixtral, Llama-3, Stable Diffusion, SDXL, SVD, and SD3, and thus requires a broad understanding of various system layers from serving API to hardware level. We achieve this by building systems that innovate new techniques as well as leveraging and contributing to open source projects including TVM, MLC-LLM, vLLM, CUTLASS, and more.
We are seeking a highly skilled and experienced Senior Machine Learning Systems Engineer to join our dynamic team. In this senior role, this person will be responsible for contributing to the latest techniques and technologies in AI and machine learning. They will lead projects, mentor junior engineers, and play a crucial role in shaping the future of our ML systems.
Responsibilities:
- Lead and improve model ingestion and robustness at the front end of our core pipeline.
- Develop cutting-edge generative AI features and drive innovation in our product offerings.
- Analyze the performance characteristics of various generative AI traffics and identify system bottlenecks.
- Develop impactful optimizations ranging from system-level to hardware-level, ensuring maximum efficiency and performance.
- Create and present groundbreaking demos to showcase our capabilities to stakeholders, customers, and the broader community.
- Mentor and guide junior engineers, fostering a collaborative and growth-oriented team environment.
- Stay up-to-date with the latest advancements in AI and machine learning, integrating new techniques and technologies into our systems.
What we look for:
- Bachelor’s, Master’s, or PhD degree in Computer Science, Electrical Engineering, or a related field.
- Extensive experience in machine learning systems engineering, with a proven track record of leading successful projects.
- Strong programming skills in C/C++ and Python with a robust computer science background.
- Experience in LLM inference systems is preferred but not required.
- Familiarity with innovative open-source projects like MLC-LLM, vLLM, TGI is a plus.
- Experience with machine learning compilers or frameworks such as TVM, MLIR, Pytorch, Tensorflow, ONNX Runtime, TensorRT is preferred.
- Excellent problem-solving skills and the ability to work effectively in a fast-paced, dynamic environment.
- Strong communication skills, with the ability to articulate complex technical concepts to both technical and non-technical stakeholders.
Diversity & Inclusion:Our mission extends beyond technological breakthroughs; we aim to mirror the diverse world we serve. We understand the unique insights, creativity, and perspectives a diverse team can bring, and we're actively seeking to broaden our horizons. As we look to the future, we aim to address our current diversity gaps and set new industry standards for inclusivity.
Benefits Our Team Enjoys: • Comprehensive Healthcare: Fully covered premiums for employees and their dependents, including Medical, Dental, Vision, Life Insurance, and Disability Insurance. • Competitive Compensation: A mix of salary, bonuses, and meaningful stock options. • Financial Benefits: Flexible Spending Accounts for healthcare and dependent care, as well as a Health Savings Account for those opting for a high deductible plan. • Future Planning: 401(k) options. • Flexible Work Options: Remote and teleworking capabilities. • Work-Life Balance: Flexible work hours, generous time off policies, company-sanctioned downtime twice a year, and company-paid holidays. • Parental Benefits: Comprehensive parental leave plans for all new parents. • Volunteer Time Off (VTO): Four days a year to give back and make a difference in communities. • Additional Leaves: Including disability, paid family medical leave, and paid military leave.
At OctoAI, we believe in creating an environment where everyone is valued and respected. Discrimination doesn't have a place in our culture. We evaluate individuals based on their abilities and effectiveness. We're committed to equal opportunity and ensuring that all qualified applicants receive consideration, free from biases related to race, color, religion, gender, identity, orientation, national origin, genetics, disability, age, or veteran status.
* Salary range is an estimate based on our AI, ML, Data Science Salary Index 💰
Tags: APIs Computer Science Engineering Generative AI LLaMA LLMs Machine Learning ML infrastructure Model deployment ONNX Open Source PhD Python PyTorch Stable Diffusion TensorFlow TensorRT vLLM
Perks/benefits: Career development Competitive pay Equity / stock options Flex hours Flexible spending account Flex vacation Health care Insurance Medical leave Parental leave Salary bonus Startup environment
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 Research Scientist jobs
- Open Data Science Manager jobs
- Open Junior Data Analyst jobs
- Open Business Data Analyst jobs
- Open Principal Data Scientist jobs
- Open Data Scientist II jobs
- Open BI Analyst jobs
- Open Sr Data Engineer jobs
- Open Business Intelligence Engineer jobs
- Open Sr. Data Scientist jobs
- Open Data Science Intern jobs
- Open Senior Business Intelligence Analyst jobs
- Open Software Engineer, Machine Learning jobs
- Open Lead Data Analyst jobs
- Open Azure Data Engineer jobs
- Open Junior Data Scientist jobs
- Open Manager, Data Engineering jobs
- Open MLOps Engineer jobs
- Open Marketing Data Analyst jobs
- Open Data Analytics Engineer jobs
- Open Data Engineer III jobs
- Open Data Engineering Manager jobs
- Open Junior Data Engineer jobs
- Open Data Analyst II jobs
- Open Product Data Analyst jobs
- Open Tableau-related jobs
- Open Data quality-related jobs
- Open Privacy-related jobs
- Open Excel-related jobs
- Open ML models-related jobs
- Open Data pipelines-related jobs
- Open APIs-related jobs
- Open PhD-related jobs
- Open PyTorch-related jobs
- Open Finance-related jobs
- Open LLMs-related jobs
- Open Data visualization-related jobs
- Open TensorFlow-related jobs
- Open Deep Learning-related jobs
- Open Consulting-related jobs
- Open Business Intelligence-related jobs
- Open Generative AI-related jobs
- Open CI/CD-related jobs
- Open NLP-related jobs
- Open Data governance-related jobs
- Open DevOps-related jobs
- Open Kubernetes-related jobs
- Open Git-related jobs
- Open Docker-related jobs
- Open Hadoop-related jobs