Software Engineer - AI/ML

We're looking for a passionate Mid-Level AI/ML Software Engineer with 2-3 years experience to join TEKHQS in Lahore. You'll work on exciting AI and machine learning projects, helping build and deploy innovative solutions in deep learning and generative AI.

Requirements

• Build and train AI models, including transformer-based models (e.g., BERT, GPT, or similar), using GPU-based environments.
• Support the development of GenAI systems, including tasks like data preprocessing, model training, and fine-tuning.
• Experiment with model optimization techniques, such as quantization or efficient attention mechanisms.
• Help implement state-of-the-art AI methods from research papers or open-source frameworks.
• Manage parts of the model lifecycle: data preparation, training, evaluation, and deployment.
• Work with multi-modal data (e.g., text, images) to support cross-domain AI applications.
• Collaborate with senior engineers to optimize models for performance, including low-latency inference and memory efficiency.
• Write clean, modular Python code and contribute to CI/CD pipelines for model deployment.

Benefits

• Generous Paid Time Off