Data Scientist

About the role

Every freight train that arrives on time and every passenger trip that ends safely depends on people, equipment, and data working together. At Wi‑Tronix, the models you train here will run on real locomotives in real time, helping crews and customers avoid the incidents that never make the news. Wi‑Tronix is the digital platform the rail industry turns to for real‑time IoT data, AI‑powered analytics, and intelligent video. This is not a typical SaaS environment, and it is not a research lab. Our work is connected to critical infrastructure, our 2026 AI roadmap is funded and shared across the company, and quality is a standard we hold each other to.

Responsibilities

• Own the full lifecycle of deep learning models that run on locomotives across North America, working with real video and sensor data and shipping to edge hardware where lighting, weather, and equipment variability are constant factors.
• Design and train custom deep neural networks for computer vision, object detection, and time series problems.
• Build and refine the labeling and training pipelines that feed model development.
• Optimize models for edge deployment using PyTorch, NVIDIA TensorRT, and ONNX.
• Create the APIs and automated testing that move models into production and keep them healthy.
• Partner with ML engineers, data engineers, and product, and translate model behavior for non‑technical stakeholders.

Requirements

• BS or MS in Computer Science, Mathematics, Statistics, or a related field.
• 3+ years building and deploying deep learning models on real data, with strong Python and PyTorch fluency.
• Computer vision fundamentals, SQL for data access, and experience with regression, classification, and object detection.
• Solid foundation in memory model, CPU and GPU architectures, and modular software design; comfort with AWS S3 and Glue.
• Agile mindset comfortable with rapid prototyping, feedback loops, and shipping work the team can act on.

Skills

• Python, PyTorch.
• Computer vision, object detection, time series.
• SQL.
• Experience with regression, classification.
• Knowledge of CPU/GPU architectures, memory model.
• AWS S3, Glue.
• Bonus: NVIDIA TensorRT, ONNX, YOLO or Detectron2, C or C++ for edge integration, Docker, IoT or safety‑critical background.

Benefits

• Full lifecycle ownership, a named career path to Senior Data Scientist, Principal Data Scientist, and AI Manager.
• Medical, dental, and vision with HSA (employer contribution).
• Employer‑paid life, AD&D, and disability coverage.
• 24/7 telemedicine.
• Employee Assistance Program.
• Financial wellness resources.

Pay

Anticipated base salary range: $103,000 to $146,000 (final compensation based on experience, qualifications, and location).

Schedule

Hybrid work environment in Bolingbrook, IL: Tuesday, Wednesday, and Thursday in office; Monday and Friday worked remotely or on site as business needs require. Ability to board a locomotive for field research. Occasional domestic and international travel may be required.