AI/ML Engineer

We Are Hiring for our Client!

Role Mission

Deliver measurable improvements to AI feature quality, reliability, and latency by owning end to-end ML workflows and collaborating with full-stack developers, data engineers, and product managers.

Core Responsibilities

• Define problem framing, labels, metrics, and guardrails for model development.
• Research, prototype, and productionize models (classical ML + deep learning,
• LLMs/RAG where applicable).
• Conduct feature engineering, training, hyperparameter tuning, and error analysis.
• Package and serve models using FastAPI/Flask, Docker/K8s, and CI/CD workflows.
• Implement observability: drift detection, quality dashboards, and automated alerting.
• Collaborate with platform engineers on scalable deployments and performance
• optimization.
• Contribute to documentation, code reviews, and knowledge sharing across teams.

Required Skills

• Python; ML frameworks (PyTorch, TensorFlow, scikit-learn).
• Data tooling: Pandas, NumPy, Arrow, SQL.
• Experiment tracking (MLflow, Weights & Biases).
• API-driven ML service development.
• Cloud deployment (AWS/GCP/Azure).
• Evaluation metrics: precision/recall, ROC-AUC, F1, MAE.
• Helpful Skills
• LLMs, Hugging Face, embeddings, vector DBs (FAISS, Pinecone, Weaviate).
• Data streaming/batch tools (Kafka, Airflow).
• Internal developer tools or dashboards.

Tools / Stack

Python, PyTorch/TensorFlow, Hugging Face, scikit-learn, FastAPI, Docker/K8s, AWS/GCP,

SQL, MLflow/W&B, vector DBs.

Experience: 2-3 Years 
Salary Range:150,000 - 170,000
Location: DHA Phase 8 Lahore
Time: 12:00PM - 8:00PM
qaseem@astraprotocol.com