Machine Learning (ML) Engineer (Senior) - Arbisoft

Our client Arbisoft is looking for Senior ML Engineer in **Lahore.
Job Description:Arbisoft **is looking for an experienced ML Engineer to design and deploy cutting-edge AI solutions, including LLMs, RAG pipelines, and agentic workflows. The ideal candidate brings deep expertise in Python, transformers, and scalable cloud-based ML systems.

Key Responsibilities:

• Design, implement, and evaluate ML/DL models using PyTorch, TensorFlow, or similar frameworks.

• Build and optimize LLM-based systems, including prompt-tuning, fine-tuning, and adapter-based training (e.g., LoRA, QLoRA).

• Can develop robust and scalable RAG pipelines. In-depth knowledge of embeddings and can work with vector databases like FAISS, Pinecone, Weaviate, etc.

• Construct and maintain Agentic AI workflows involving multi-step reasoning, tool calling, memory components, and planning logic.

• Work with Proprietary APIs, as well as open-source libraries and models

• Develop modular and clean Python code, adhering to software engineering best practices (OOP, reusable components, testing).

• Implement scalable solutions in cloud environments (like AWS), leveraging GPU/TPU resources effectively.

• Design inference pipelines that are robust and optimized for latency and throughput.

• Collaborate with research and product teams to translate ideas into production-grade ML features.

Required Skills:

• 5+ years of experience in machine learning and deep learning, including building models from scratch.

• Has a track record of shipping ML solutions that scale in production.

• Strong proficiency in Python and deep understanding of software design principles.

• Proven experience with transformer-based architectures, LLMs, and embedding models.

• Hands-on experience with RAG systems, deep understanding of agent-based systems. Familiarity with LangChain, LlamaIndex, or similar frameworks.

• Experience with cloud platforms (AWS/GCP/Azure) and understanding of scalability, resource optimization, and model deployment.

• Familiarity with performance profiling, efficient model serving, and hardware-aware design (e.g., GPU utilization, quantization).

• Ability to read, debug, and contribute to complex ML/DL codebases.

Good to have:

• Experience with MLOps, orchestration tools (e.g., Airflow, AWS Step Functions), containerization (Docker, Kubernetes).

• Exposure to optimization toolkits (ONNX, TensorRT) and serving frameworks (Triton, TorchServe).