Data Scientist

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

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. We are hiring a Data Scientist who wants their models to leave the notebook and live on the rail.

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.

THE ROLE

You will 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

Who Thrives Here

You are a methodical experimenter who keeps careful logs and designs clean ablations. You would rather see a smaller model run well on real hardware than a bigger one win a leaderboard.

• 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
• Bonus: NVIDIA TensorRT, ONNX, YOLO or Detectron2, C or C++ for edge integration, Docker, IoT or safety critical background

What We Offer

A specific and real mission with locomotive video and sensor data nobody else has access to. Full lifecycle ownership, a named career path to Senior Data Scientist, Principal Data Scientist, and AI Manager, and benefits that back your whole life: medical, dental, and vision with HSA (employer contribution), employer-paid life, AD&D, and disability coverage, 24/7 telemedicine, an Employee Assistance Program, and financial wellness resources.

WORK ENVIRONMENT

Hybrid in Bolingbrook, IL: Tuesday, Wednesday, and Thursday in office, with 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.

READY TO SEE YOUR MODELS RUN IN THE REAL WORLD?

If you want your next role to be the one where your models actually ship and the outcomes are measured in safer crews and more reliable service, we would like to meet you. Bring your curiosity, your craft, and your standards. We will bring the mission, the data, and a team that trusts each other.

Equal Opportunity Employer

Applicants must be legally authorized to work in the United States. This position does not offer employer‑based visa sponsorship now or in the future. This position’s anticipated base salary range is $103,000 to $146,000. Final compensation will be based on experience, qualifications, and location. Wi‑Tronix is proud to be an Equal Opportunity, Aff… E‑Verify, and Drug Free Employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, gender, gender identity or expression, sexual orientation, national origin, physical or mental disability, age, or veteran status or any other basis protected by federal, state, or local law. If you need reasonable accommodation for any part of the application process due to disability, please inform us at [e-mail] so that we can discuss the appropriate alternatives available. Wi‑Tronix does not review unsolicited resumes from recruiters or search firms.