[Remote] Sr/Staff AI Engineer (Remote - US) Job Details | BNSF
Note The job is a remote job and is open to candidates in USA. BNSF Railway is one of the largest U.S. railroads, dedicated to transporting freight across the nation. They are seeking a Senior/Staff AI Engineer to design and implement AI models for railroad operations, focusing on predictive maintenance and safety monitoring while collaborating with cross-functional teams.
Responsibilities
- 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
- 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
- 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
- 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
- Monitor deployed models and systems for performance, scalability, and accuracy, ensuring continuous improvement and reliability
- Troubleshoot issues, retrain models, and implement necessary updates
- 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
- 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 Skills
- Authorized to work in the US
- Minimum 6 years experience in AI/ML development, data science, software engineering, or relevant experience
- 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
- Ability to interpret complex datasets and derive actionable insights
- Proven ability to tackle real-world challenges using AI and machine learning techniques
- Excellent verbal and written communication skills for working with cross-functional teams and stakeholders
- 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 Benefits
- 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.
- Annua Apply To this Job