AI/ML Developer

The Judge Group; Pleasanton (CA); On-site; USD 180,000 - 240,000; Full time

Benefits offered by this job

Medical, dental, vision coverage

Retirement savings program

Paid time off

Wellness resources

Continuous learning and career growth

Job summary

The Judge Group is seeking an AI/ML Developer to design, build, and scale next-generation AI solutions for the food supply chain. You will develop production-grade ML models, build scalable data pipelines, and collaborate with product teams to deliver AI-powered products from concept to production.

Ideal candidates have 5+ years of experience deploying ML solutions, strong software engineering fundamentals, and hands-on experience with cloud-native environments, containers, and CI/CD practices.

Qualifications

• Bachelor's or Master's in CS/Data Science or related field or equivalent experience.
• 5+ years of experience building and deploying ML solutions in production.
• Ability to translate complex business problems into ML solutions.
• Strong foundation in data science methods: supervised/unsupervised learning, time-series forecasting, classification, clustering.

Responsibilities

• Design, develop, and deploy scalable ML models and data pipelines.
• Solve problems in supply chain optimization, forecasting, and anomaly detection.
• Collaborate with product, software, and data science teams from ideation to production.

Skills

Supervised learning

Unsupervised learning

Time-series forecasting

Classification

Education

Bachelor's degree in CS/Data Science (or related)

Master's degree preferred

Tools

TensorFlow

PyTorch

Scikit-learn

XGBoost

Docker

Kubernetes

BigQuery

Spark

Airflow

dbt

Job description

Our client is currently seeking a AI/ML DeveloperAI/ML Software EngineerAbout the Role

Are you a data-driven engineer who sees patterns where others see complexity? Do you enjoy building intelligent systems that solve real-world business challenges? We are seeking an experienced AI/ML Software Engineer (P3) to help design, build, and scale next-generation AI solutions that drive smarter decision-making across the food supply chain.

In this role, you will develop and deploy machine learning models, create scalable data pipelines, and collaborate cross-functionally to deliver AI-powered products from concept through production. This is an opportunity to work with modern AI technologies and apply them to high-impact business problems at scale.

What You'll Do

Build Intelligent Systems

• Design, develop, and deploy scalable machine learning models and data pipelines.
• Create solutions for supply chain optimization, demand forecasting, and anomaly detection.
• Ensure models are interpretable, reliable, and production-ready.

AI-Powered Product Development

• Partner with Product Managers, Software Engineers, and Data Scientists to integrate machine learning solutions into customer-facing products.
• Drive projects through the full lifecycle, from ideation to production deployment.

Data Engineering & Analytics

• Work with large-scale datasets sourced from transactional systems, IoT sensors, and third-party providers.
• Build and maintain data pipelines using technologies such as BigQuery, Airflow, Spark, and dbt.
• Perform data exploration, feature engineering, and transformation activities.

Model Development & Optimization

• Develop and train machine learning and deep learning models using Python, TensorFlow, PyTorch, XGBoost, and Scikit-learn.
• Conduct model evaluation, experimentation, A/B testing, and performance benchmarking.
• Continuously optimize models for accuracy, scalability, and efficiency.

MLOps & Deployment

• Deploy machine learning solutions using Docker, Kubernetes, and CI/CD pipelines.
• Monitor model performance, drift, and reliability in production environments.
• Build robust MLOps frameworks that support model versioning, reproducibility, and automated retraining.

Innovation & Continuous Improvement

• Stay current with emerging AI/ML technologies and industry trends.
• Evaluate and implement new tools and approaches that create measurable business value.

Required Qualifications

• Bachelor's or Master's degree in Computer Science, Data Science, Machine Learning, or a related field (or equivalent practical experience).
• 5+ years of experience developing and deploying machine learning solutions in production environments.
• Proven ability to translate complex business challenges into effective machine learning solutions.
• Strong expertise in:
  • Supervised and unsupervised learning
  • Time-series forecasting
  • Classification
  • Clustering
• Solid software engineering fundamentals, including:
  • Version control
  • Automated testing
  • Documentation
  • AI/ML system architecture
• Experience working in cloud-native environments with containerization and CI/CD practices.
• Strong communication skills with the ability to explain complex technical concepts to technical and non-technical audiences.
• Experience supporting real-time inference, large-scale batch predictions, and model retraining workflows.

Preferred Qualifications

• Experience with:
  • Optimization algorithms
  • Recommendation systems
  • Large Language Models (LLMs)
  • Graph Neural Networks (GNNs)
• Experience applying AI solutions within supply chain, logistics, forecasting, or operations environments.

Technical SkillsProgramming & Machine Learning

• Python
• SQL
• TensorFlow
• PyTorch
• Scikit-learn
• XGBoost

Data Engineering

• Oracle
• BigQuery
• Apache Spark
• Apache Airflow
• Pandas
• dbt

MLOps & DevOps

• MLflow
• Docker
• Kubernetes
• GitHub Actions
• Terraform

Cloud Platforms

• Google Cloud Platform (GCP)
  • BigQuery
  • Vertex AI
  • Dataflow
  • Cloud Functions
• Amazon Web Services (AWS)
  • SageMaker

AI Productivity Tools

• GitHub Copilot
• ChatGPT
• Cursor AI
• Otter AI

Analytics & Visualization

• Looker
• Plotly
• Dash
• Jupyter Notebooks

Additional Technologies

• REST APIs
• Feature Stores
• Microservices Architecture

What Makes You SuccessfulAI-First Mindset

You naturally look for opportunities to embed intelligence into solutions rather than treating AI as an afterthought.

Bias for Action

You thrive in fast-paced environments, embrace ambiguity, and adapt quickly to evolving technologies and business needs.

Ownership Mentality

You proactively identify opportunities, solve problems independently, and take accountability for outcomes.

Trust Builder

You earn credibility through execution, collaboration, and a commitment to delivering high-quality work.

Why Join Us?

• Work on high-impact AI initiatives that directly influence business decision-making.
• Build production-grade machine learning systems used at scale.
• Collaborate with talented engineers, product leaders, and data scientists.
• Enjoy autonomy, ownership, and the opportunity to shape the future of AI-powered products.
• Stay at the forefront of machine learning innovation while solving meaningful real-world problems.

Our client offersa comprehensive benefits package designed to support your health, financial well-being, professional growth, and work-life balance. Employees enjoy competitive medical, dental, and vision coverage, retirement savings programs, paid time off, wellness resources, and opportunities for continuous learning and career development depending on location and position.