ML Pipeline Engineer

We are looking for an ML Pipeline Engineer to take ownership of and enhance our continuous integration (CI) pipeline for machine learning. Our pipeline currently automates model retraining using data from MySQL, Snowflake, and BLOB storage, performs model performance checks, and ensures the integrity of our models. Now, we’re looking to extend its capabilities by improving model deployment and further optimizing automation.

This role requires a deep understanding of machine learning workflows, infrastructure, and automation.

RESPONSIBILITIES

• Build and maintain robust, scalable pipelines for data preprocessing, training, and inference.
• Automate machine learning workflows, including feature extraction, model training, and evaluation.
• Deploy machine learning models into production environments, ensuring high availability and performance.
• Monitor production systems for model drift, performance issues, and data quality.
• Collaborate with Data Scientists to integrate models into production environments effectively.
• Optimize infrastructure for cost, performance, and scalability, using cloud platforms and distributed systems.
• Implement CI/CD pipelines for machine learning workflows to enable faster and more reliable deployments.
• Ensure compliance with data governance, security, and regulatory standards in production systems.

• Bachelor’s degree in Computer Science, Data Engineering, or a related field, or equivalent experience.
• 3+ years of experience in machine learning engineering, software engineering, and DevOps.
• Strong programming skills in Python and experience with ML libraries.
• Proficiency in cloud platforms and their ML services.
• Hands-on experience with containerization (Docker) and orchestration tools (Kubernetes) is a plus.
• Knowledge of distributed data processing frameworks (e.g., Apache Spark, Hadoop).
• Strong understanding of CI/CD practices and tools for ML pipelines.

PREFERRED QUALIFICATIONS

• Experience with monitoring tools for tracking model performance and system health.
• Familiarity with workflow orchestration tools.
• Knowledge of MLOps tools.
• Exposure to ML model training and deployment—and a strong desire to get hands-on with data science tasks.

TOOLS AND TECHNOLOGY

• Programming: Python, Bash scripting.
• ML Frameworks: PyTorch, Scikit-learn, XGBoost.
• Infrastructure: Docker, Kubernetes, Terraform.
• Cloud Platforms: Azure ML, Azure Function Apps

SOFT SKILLS

• Strong attention to detail and focus on reliability.
• Effective collaboration with cross-functional teams, including Data Scientists and DevOps.
• Problem-solving mindset with the ability to debug complex issues.
• Excellent organizational skills to manage and prioritize tasks effectively.

About OneRail

OneRail, headquartered in Orlando, FL, is the emerging leader in Final Mile delivery technology. Our OmniPoint SaaS platform uses proprietary algorithms and artificial intelligence to power efficient same-day delivery programs for major companies like Lowe’s, Pepsi, American Tire Distributors, and Advance Auto Parts.

OmniPoint is the cornerstone of OneRail’s integrated final mile delivery solution.

• Capture. Within moments of a purchase that requires same-day delivery, OmniPoint captures a shipper’s “demand signal” from their POS, ERP, or e-commerce system: what was ordered, where it is, where it’s going, and when it needs to be there.
• Optimize. OmniPoint uses this data to specify the optimal vehicle for the order’s size and weight, as well as any extra services required. OmniPoint then calculates all-in shipping costs and optimizes delivery routing.
• Execute. Customers can direct deliveries to their fleet, or OneRail’s nationwide network of 12 million couriers available 24/7/365.
• Support. Deliveries can be supported as they occur by OneRail’s skilled Exceptions Assist team to handle unforeseen delivery issues on the customer’s behalf.
• Follow Up. OmniPoint enables the consumer or receiving business to track their order delivery in real time and provide feedback through a customer-branded interface.

OneRail has been recognized by Gartner as one of the most innovative solutions in the Final Mile delivery space. We ranked in the top 100 of the Inc. 5000 list of the fastest-growing US private companies in 2022 and 2023 and received an Inc. Best Workplace award in 2023.

Regular, Full-time Team Member Benefits

• Competitive base compensation
• Health and wellness Insurance with generous company contribution (medical, dental, vision)
• Company-paid life insurance, short-term and long-term disability
• 4% 401K match with immediate vesting of Company match
• Continuing Education Opportunities
• Flexible/Open (Uncapped) Paid Time Off (PTO) Policy for Salaried Team Members*
• Competitive Paid Time Off (PTO) Policy for Hourly Team Members*
• Generous Company-paid Holidays

Onsite Team Member Job Perks

• Relaxed/casual work environment
• Standing desks
• Recreational and Video games
• Large breakroom and lounge
• Stocked kitchen and fridges
• Cappuccino machine
• Onsite restaurant and daily food trucks
• Offices conveniently located with quick access to I-4, SR-528, and Florida's Turnpike
• Ample free parking

OneRail's Travel Policy for Remote Team Members

While OneRail allows remote work for certain posted roles, noted as "Remote Work Available" on the Company's website, a core requisite for all roles is that team members be willing and able to occasionally travel to OneRail's Orlando, Florida headquarters. All team members are required to participate in on-site onboarding at the onset of employment and then on an as-needed basis thereafter, as a condition of their employment.

To view more details about what is like to work at OneRail, or to view a full list of career opportunities, please visit https://www.onerail.com/careers/