Staff Machine Learning Engineer

Posted 3 months agoViewed

View full description

💎 Seniority level: Staff, 5+ years ML Experience

📍 Location: USA

💸 Salary: 235000 USD per year

🔍 Industry: E-commerce / AI

🏢 Company: Wizard👥 11-50 Customer Service Manufacturing

⏳ Experience: 5+ years ML Experience

🪄 Skills: Node.jsPythonFlaskGCPKubernetesMachine LearningPyTorchAirflowFastAPIPandas

Requirements:

MS/Ph.D. in computer science, mathematics, or another quantitative discipline or exceptional work experience in ML engineering.
5+ years ML Experience.
Experience developing machine learning-driven products at scale.
Experience solving problems using Machine Learning with PyTorch.
Exceptional understanding of designing ML systems in production at scale.
Practical experience with large data systems, data models, and data pipelines.
Ability to learn quickly on tech stack including GCP, Kubernetes, Airflow, Pandas, PyTorch, Python, Node.js.
Extensive experience with model deployment using Flask, FastAPI, or similar frameworks.

Responsibilities:

Build and maintain a training pipeline for fine-tuning open-source LLMs.
Test different LLM architectures at varying sizes and quantizations.
Serve and deploy the LLM on GPUs, avoiding OOM at runtime.
Formulate product issues as ML problems and quickly iterate on solutions.
Triage customer issues related to model/system shortcomings.
Simplify complex architecture communications for internal stakeholders.
Stay updated with the generative AI field and integrate new developments.

Apply

Related Jobs

Apply

🔥 (Senior or Staff) Machine Learning Engineer

Posted 2 days ago

📍 United States, Canada

🧭 Full-Time

💸 180000.0 - 210000.0 USD per year

🔍 Software Development

🏢 Company: Gretel👥 51-100💰 $52,199,996 Series B over 3 years agoArtificial Intelligence (AI)Data Collection and Labeling Machine Learning Privacy Generative AI Information Technology Software

🔧 Requirements

5+ years of professional experience in software development
Significant experience with Python's ML ecosystem
Strong foundation in deep learning and related mathematics

💡 Responsibilities

Design and implement the software behind the core product
Collaborate with applied scientists and engineers
Unlock real-world use cases for synthetic data generation

PythonMachine LearningMLFlowPyTorchTensorflow

Posted 2 days ago

Apply

🔥 Staff Machine Learning Engineer

Posted 9 days ago

📍 United States, Canada

💸 225000.0 - 250000.0 USD per year

🏢 Company: Inspiren👥 11-50💰 $2,720,602 over 2 years agoMachine Learning Analytics Information Technology Health Care

🔧 Requirements

8+ years of professional experience in machine learning, software engineering, or a similar domain
5+ years with machine learning algorithms, model development, and model deployment
3+ years in advanced algorithm design and analysis
Proficiency in Python, R, or Scala languages for data science
Experience with TensorFlow, Keras, or PyTorch for neural network modeling
Ability to optimize machine learning models for performance
Proven track record of deploying machine learning models into production

💡 Responsibilities

Design and develop advanced algorithms to address complex business challenges.
Optimize machine learning models to achieve faster speed and greater efficiency.
Evaluate machine learning models to ensure they deliver accurate and relevant results.
Provide technical guidance and mentor junior machine learning engineers.

PythonKerasMachine LearningNumpyPyTorchAlgorithmsData scienceTensorflowSoftware Engineering

Posted 9 days ago

Apply

🔥 Staff Machine Learning Engineer

Posted 25 days ago

📍 Canada, Latin America, Alabama, Arizona, California, Colorado, Connecticut, Florida, Georgia, Illinois, Indiana, Massachusetts, Minnesota, Nevada, New Jersey, New York, North Carolina, Oregon, Pennsylvania, Rhode Island, Tennessee, Texas, Utah, Virginia, Washington

💸 140000.0 - 220000.0 USD per year

🔍 Retail AI

🏢 Company: Lily AI

🔧 Requirements

10+ years in building large-scale machine learning solutions and ML Ops practices.
Working with LLM APIs and serving LLMs in-house at scale.
Proficiency in Kubernetes, RDBMS, and API-driven development.
Experience in model serving in low-latency, high-throughput use cases.
Knowledge of observability, data pipeline design, service scaling, and cost optimization.
Strong emphasis on code hygiene, including review, documentation, testing, and CI/CD practices.
Proficiency in Python and PyTorch.
Extensive experience with the scientific Python ecosystem.
Proficiency in cloud-native application development.
Action-oriented with the ability to articulate complex concepts.

💡 Responsibilities

Define, design, and maintain scalable Machine Learning data pipelines, training infrastructure, and inference systems.
Optimize, benchmark, and productionize deep learning models to extract high-value product attributes.
Drive cost efficiency and throughput improvements, owning relevant KPIs.
Promote and implement software engineering best practices across the team.
Shape and evolve the technical stack to meet business and technical needs.
Transition research prototypes into robust, production-ready systems.
Deploy, monitor, and continuously improve models in production environments.
Optimize model performance, focusing on memory usage and latency.
Automate workflows by building efficient pipelines and orchestration frameworks.
Develop tools and shared libraries to boost team productivity.

PythonKubeflowKubernetesMachine LearningMLFlowPyTorchAzure

Posted 25 days ago

Apply

🔥 Senior Staff Machine Learning Engineer, Ads Optimization

Posted 2 months ago

📍 United States, Canada

🧭 Full-Time

💸 260800.0 - 365100.0 USD per year

🔍 Digital Advertising

🏢 Company: Reddit👥 1001-5000💰 $410,000,000 Series F over 3 years ago🫂 Last layoff over 1 year agoNews Content Social Network Social Media

🔧 Requirements

10+ years of contributing high-quality code to production systems that operate at scale.
2+ years of experience operating as a Senior Staff engineer.
5+ years of experience in building control systems, PID controllers, multi-armed bandits, reinforcement learning algorithms, or bid/pricing optimization systems.
Experience leading large engineering teams and collaborating with cross-functional partners.
Experience designing optimization algorithms in an ad serving platform or other marketplaces is preferred.
Familiarity with control systems and reinforcement learning algorithms is a strong plus.
Significant experience in Java, Python, Go, Scala, C++, or similar languages.
Experience with data processing frameworks like Spark, Flink, Kafka.
Familiarity with AWS or GCP as a cloud service provider.
Experience with tools like Kubernetes, Docker, Airflow, etc.
Familiarity with datastores such as ElasticSearch, Redis, and Postgres.

💡 Responsibilities

Building Reddit-scale optimizations to improve advertiser outcomes using innovative techniques.
Leveraging live auction data and model predictions to adjust campaign bids in real-time.
Incorporating knowledge of the Reddit ads marketplace into budget pacing algorithms.
Leading a team of 15+ engineers on designing new bid & budget optimization products.
Conducting rigorous A/B experiments to evaluate business impact.
Collaborating with cross-functional teams for customer representation.

AWSDockerPythonElasticSearchGCPJavaKafkaKubernetesMachine LearningPyTorchC++AirflowAlgorithmsCassandraGoPostgresRedisSparkTensorflowScala

Posted 2 months ago

Apply

🔥 Staff Machine Learning Engineer, Gen AI

Posted 3 months ago

📍 US

🧭 Full-Time

💸 165000.0 - 283000.0 USD per year

🔍 Technology

🏢 Company: Mozilla👥 5001-10000💰 $300,000 Angel about 20 years ago🫂 Last layoff 4 months agoInternet Open Source Web Browsers Software Browser Extensions

🔧 Requirements

6+ years experience as a Machine Learning engineer building tooling and services for ML applications in production.
Experience with designing and building ML tooling and infrastructure for training, deploying, inference, and validation of models.
Experience with distributed systems and platforms for AI integrations.
Strong problem-solving skills and the ability to communicate complex concepts effectively.
Experience working collaboratively with product managers and non-engineering teams.
Effective documentation and communication skills.

💡 Responsibilities

Lead the design, development, and integration of Generative AI solutions in Firefox.
Collaborate cross-functionally with product management, full-stack engineering, and design.
Build infrastructure for training and inference of LLMs and small models.
Implement robust validation and testing procedures.
Continuously monitor and optimize deployed models for performance and efficiency.

AWSArtificial IntelligenceCloud ComputingKubernetesMachine LearningPyTorchAlgorithmsData engineeringData scienceTensorflowCommunication SkillsCI/CDDocumentation

Posted 3 months ago

Apply

🔥 Staff Machine Learning Engineer

Posted 3 months ago

📍 United States, Canada

🧭 Full-Time

💸 199000.0 - 244000.0 USD per year

🔍 Aerospace

🏢 Company: Acubed👥 51-100 Innovation Management Aerospace Manufacturing

🔧 Requirements

Master's degree or equivalent experience in computer science
11+ years in software/system design for aerospace or robotics
6+ years as a CV/ML perception software engineer
Experience with deep learning frameworks like PyTorch and TensorFlow
Experience deploying ML models to production
Extensive experience in Python
Strong communication skills

💡 Responsibilities

Design and develop ML models for vision-based flight capabilities
Drive technical direction and team guidance
Evaluate ML models for effectiveness and efficiency
Write maintainable product-level code
Collaborate with multiple teams across Airbus
Prototype projects using the latest innovations

PythonMachine LearningPyTorchTensorflow

Posted 3 months ago

Apply

🔥 Staff Machine Learning Engineer

Posted 4 months ago

📍 San Francisco, CA, Washington, DC

🧭 Full-Time

💸 180000 - 250000 USD per year

🔍 Artificial Intelligence

🏢 Company: primer.ai

🔧 Requirements

BS, MS, or PhD degree in computer science, related field or equivalent practical experience.
5+ years as a backend software engineer integrating and deploying ML-driven functionality.
Mastery of data structures and algorithms, able to use them practically.
Experience with machine learning algorithms and tools such as Numpy, PyTorch, TensorFlow; bonus for experience with LLMs and NLP.
Experience authoring APIs in Python or Golang.
Curiosity, enthusiasm, and a love for teaching and learning.

💡 Responsibilities

Understand the landscape of NLP problems and help make strategic investments.
Design & develop distributed architectures, libraries, and systems impacting the company.
Integrate with features like event and topic detection, relationship extraction, etc.
Collaborate with cross-functional teams to drive technology vision and roadmap.
Triage and debug issues, identifying their sources and impacts.
Improve and promote engineering best practices across the company.

LeadershipPythonArtificial IntelligenceDesign PatternsMachine LearningNumpyPyTorchAlgorithmsData scienceData StructuresGoTensorflow

Posted 4 months ago

Apply