Sr/Staff AI Engineer

The posted range for this position represents the amount bnsf | tech reasonably expects to pay for the position based on the level, scope, and responsibilities of the role

Individual compensation and level of position offered is determined by the hiring location and additional factors including but not limited to job-related skills, experience, and relevant education or training

In addition to base pay and bonus eligibility, BNSF offers a comprehensive benefits package

This is a full-time remote position

Employees may work from anywhere within the contiguous 48 states of the United States Travel is up to 20%

Employees will be required to occasionally travel to our corporate headquarters in Fort Worth, TX for in person meetings

Travel expenses for business needs will be covered by BNSF This position is open to candidates who are currently authorized to work in the United States

We are also open to sponsoring H-1B transfers, TN nonimmigrant status, and STEM OPT candidates with at least 2 years of remaining eligibility

Apply early as this job may be removed or filled prior to the closing date, which is approximately seven (7) days after the posting date. data & ai: lead our charge into the future as an ai company by transforming our data assets into a real time enterprise

Key responsibilities may include: AI Development and Implementation: Assist in designing, developing, and implementing AI models and algorithms tailored to railroad operations, including predictive maintenance, scheduling, and safety monitoring

Build scalable AI systems to analyze real-time data from rail infrastructure, rolling stock, and IoT devices

Data Collection and Analysis: Collect, preprocess, and clean data from various sources (e.g., sensors, GPS, maintenance logs) for training AI models

Conduct exploratory data analysis to identify trends and patterns relevant to rail operations

Build and Maintain Extract, Transform, Load (ETL) pipelines for AI applications

Optimization and Automation: Fine tune and optimize algorithms to streamline train scheduling, route planning, and fuel efficiency

Support the automation of operational processes and decision-making using AI-powered tools

System Integration and Deployment: Collaborate with software engineers and DevOps teams to integrate AI solutions into production environments

Ensure AI systems are robust, reliable, and compliant with railroad industry standards

Performance Monitoring and Improvement: Monitor deployed models and systems for performance, scalability, and accuracy, ensuring continuous improvement and reliability

Troubleshoot issues, retrain models, and implement necessary updates

Collaboration and Communication: Work closely with cross-functional teams, including data scientists, software engineers, business analysts, to understand their needs and enable AI-driven solutions

Communicate technical concepts and project updates to both technical and non-technical stakeholders

Research and Innovation: Stay updated on emerging AI technologies and trends to identify opportunities for innovation in rail transportation

Conduct research and propose new methodologies for solving complex railroad challenges

The duties and responsibilities in this posting are representative categories to be used in deciding whether to apply for this position

This is not an exhaustive list of the position’s duties

At BNSF Railway, we encourage individuals from all backgrounds to apply, showcasing their skills, experiences and development

We provide resources and tools to help you reach your full potential, fostering a supportive and inclusive environment

Basic Qualifications Authorized to work in the US

Minimum 6 years experience in AI/ML development, data science, software engineering, or relevant experience

Technical Skills: Strong in programming languages such as Python, R, or Java

Experience with machine learning frameworks (e.g., TensorFlow, PyTorch, scikit-learn) and deep learning architectures

Familiarity with data engineering tools and platforms (e.g., Hadoop, Spark, EMS, AWS S3, PySpark, RDS, JupyterLab) and cloud services (AWS, Azure, Databricks) and open source environments Strong knowledge of databases (SQL and NoSQL) Strong knowledge of Mlops ecosystem Strong understanding of generative AI models Strong in Data Pipelines for GenAI (e.g., cleaning, chunking embeddings) Strong in GenAI tools and frameworks (LangChain, LangGraph) Strong in Vector Databases and embeddings Familiar with Agentic architecture Analytical Skills: Ability to interpret complex datasets and derive actionable insights

Problem-Solving: Proven ability to tackle real-world challenges using AI and machine learning techniques

Communication: Excellent verbal and written communication skills for working with cross-functional teams and stakeholders

Preferred Qualifications Able to work now and in the future without BNSF’s assistance (whether monetary, through sponsorship, or otherwise) in obtaining, maintaining, or extending employment authorization (including H-1B, STEM OPT/CPT, or TN nonimmigrant status)

Bachelor's degree or higher in Computer Science, Artificial Intelligence, Data Science, Engineering, or a related field Experience with predictive maintenance, anomaly detection, or industrial IoT applications

Knowledge of railroad operations, transportation systems, or logistics

Familiarity with MLOps practices and tools for model lifecycle management

Experience deploying AI systems in production environments with a focus on scalability and reliability

Previous experience in transportation, supply chain, or infrastructure industries

At BNSF, you will have access to a comprehensive and competitive benefits package including: An industry-leading 401(k) and renowned Railroad Retirement program

A range of robust health care options for you and your dependents (including domestic partners), including medical, dental, vision, telemedicine, mental health, cancer support, and high-quality care network options

Health care spending accounts (HSA) with employer contributions, as well as life and disability insurance, provided at no cost

Family benefits including parental, pediatric and family building support, adoption and surrogacy reimbursement, and dependent care spending account (with employer match)

Access to discounts on travel, gym memberships, counseling services and wellness support

Annual bonus (Incentive Compensation Program) Generous leave / time off policies

For more information, visit Benefits

Please be aware of potential fraud that can occur when searching for new career opportunities

Please review our FAQ for more information and awareness

All positions require pre-employment background verification

BNSF Railway is an Equal Opportunity Employer, all qualified applicants receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability or protected veteran status.