AI Backend Engineer - Python, FastAPI, LLM (Onsite, Rawalpindi, USD Salary)

Requirements:

• Bachelors degree in Computer Science, Artificial Intelligence.
• 4-6 years of experience in backend engineering, AI/ML engineering, or related roles.
• Strong expertise in Python, FastAPI, and backend API development.
• Hands-on experience with PyTorch, TensorFlow, PyTorch Lightning, and ML model deployment.
• Proficiency in LLM integration, prompt engineering, NLP, and computer vision.
• Strong SQL skills with experience in PostgreSQL and distributed data systems.
• Experience with Databricks and PySpark for large-scale data processing.
• Proficiency with AWS (S3, EC2), Docker, Linux, and CI/CD pipelines.
• Familiarity with Git, Redis, Hugging Face, OpenPose, and LiveKit.
• Strong analytical, debugging, and communication skills.

Responsibilities:

• Develop and maintain backend services and APIs using Python and FastAPI.
• Integrate LLMs (OpenAI, Anthropic APIs), perform prompt engineering, and build AI-driven workflows.
• Implement and optimize machine learning pipelines using PyTorch, TensorFlow, PyTorch Lightning.
• Work on computer vision, NLP, and model fine-tuning tasks.
• Build and manage distributed data processing workflows using PySpark and Databricks.
• Design, manage, and optimize SQL/PostgreSQL databases.
• Deploy and manage cloud services using AWS (S3, EC2) and Docker containers.
• Implement CI/CD pipelines using GitHub Actions or GitLab.
• Work with tools such as Git, Redis, Hugging Face, OpenPose, and LiveKit.
• Collaborate with cross-functional teams and ensure high-quality documentation.