Edge ML Data Scientist for Real-Time Rail Analytics

Wi-Tronix; Bolingbrook (IL); Hybrid; USD 103,000 - 146,000; Full time

Benefits offered by this job

Medical benefits

Dental benefits

Vision benefits

HSA

Life insurance

AD&D

Disability insurance

Telemedicine

Employee Assistance Program

Financial wellness resources

Job summary

Wi‑Tronix seeks a Data Scientist to own the full lifecycle of deep learning models running on locomotives across North America, handling real video and sensor data and shipping to edge hardware. You will design custom networks, refine data pipelines, and optimize models for edge deployment using PyTorch, TensorRT, and ONNX, while building production APIs and tests.

You will collaborate with ML engineers, data engineers, and product teams, translating complex model behavior for non-technical

Qualifications

• BS/MS in Computer Science, Mathematics, Statistics, or a related field.
• 3+ years building and deploying deep learning models on real data with Python and PyTorch.
• Strong computer vision fundamentals; SQL for data access; experience with regression, classification, and object detection.
• Solid foundation in memory models, CPU/GPU architectures, and modular software design; comfort with AWS S3 and Glue.
• Agile mindset with rapid prototyping, feedback loops, and shipping work for the team.
• Bonus: NVIDIA TensorRT, ONNX, YOLO or Detectron2, C/C++, Docker, IoT or safety-critical background.

Responsibilities

• Design and train custom deep neural networks for computer vision and time series problems.
• Build and refine labeling and training pipelines feeding model development.
• Optimize models for edge deployment using PyTorch, TensorRT, and ONNX.
• Create APIs and automated testing to move models into production and monitor health.
• Collaborate with ML engineers, data engineers, and product to translate model behavior for non-technical stakeholders.

Skills

Python

PyTorch

SQL

Agile

Edge deployment

Model deployment

Education

BS/MS in Computer Science or related field

Tools

NVIDIA TensorRT

ONNX

YOLO

Detectron2

Docker

AWS S3

AWS Glue

C/C++

Job description

Wi‑Tronix seeks a Data Scientist to own the full lifecycle of deep learning models running on locomotives across North America, handling real video and sensor data and shipping to edge hardware. You will design custom networks, refine data pipelines, and optimize models for edge deployment using PyTorch, TensorRT, and ONNX, while building production APIs and tests.

You will collaborate with ML engineers, data engineers, and product teams, translating complex model behavior for non-technical