docker
git
langchain
langgraph
mlflow
python
pytorch
scikit-learn
tensorflowJob Description
Job Description
Required
- 3–6 years of experience in ML, AI engineering, or a related applied role
- Strong Python skills — pipelines, model training, API development
- Hands-on experience with ML frameworks (PyTorch, TensorFlow, or scikit-learn)
- Experience building and maintaining data pipelines for ML workflows
- Practical LLM experience — prompt engineering, chaining, and agent frameworks (LangChain, LangGraph, or similar)
- Experiment tracking tools — MLflow, Weights & Biases, or equivalent
- Strong software engineering fundamentals — Git, testing, clean code, documentation
Preferred
- Experience with hardware performance data or systems-level modeling
- Familiarity with RAG systems and production ML deployment
- Knowledge of dataset versioning tools (DVC) and containerization (Docker)
- Experience building internal developer tools
Education
Bachelor's, Master's, or PhD in Computer Science, Machine Learning, or a related field — or equivalent hands-on experience.
Location: Cupertino, CA
Salary Range: $70,000-$130,000 a Year