Software Engineer - AI/ML

About the Role

Were looking for a passionate Mid-Level AI/ML Software Engineer with 2-3 years experience to join TEKHQS in Lahore. Youll work on exciting AI and machine learning projects, helping build and deploy innovative solutions in deep learning and generative AI. This role is perfect for someone eager to dive into real-world AI challenges, learn cutting-edge techniques, and contribute to our global tech solutions. Expect to work with raw data, experiment with models, and collaborate with a talented team to bring ideas to life.

Key Responsibilities

• 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.

Requirements:

Core AI/ML Skills:

• Familiarity with deep learning concepts like transformers, self-attention, or normalization techniques (e.g., LayerNorm).

• Basic understanding of token sampling (e.g., Top-k, Top-p) or training strategies like curriculum learning.

• Experience with supervised or unsupervised learning (e.g., classification, regression, or masked modeling).

Training Infrastructure:

• Hands-on experience with Python-based ML frameworks like PyTorch, TensorFlow, or HuggingFace.

• Exposure to distributed training or mixed precision (e.g., fp16) is a plus.

• Familiarity with data formats like CSV, JSON, or Parquet for model training.

Model Optimization & Serving:

• Basic knowledge of model optimization techniques (e.g., quantization or pruning).

• Exposure to model serving tools (e.g., HuggingFace Inference, ONNX, or similar) is a plus.

GenAI Systems:

• Interest in generative AI, including LLMs or Retrieval-Augmented Generation (RAG).

• Familiarity with tokenizers (e.g., Byte Pair Encoding or SentencePiece) is a bonus.

Software Engineering:

• Ability to write clean, testable Python code.

• Basic experience with version control (e.g., Git) and CI/CD pipelines.

• Familiarity with Docker or profiling tools (e.g., PyTorch Profiler) is a plus.

Mathematical Foundations:

• Understanding of core concepts like gradient descent, loss functions (e.g., cross-entropy), or basic linear algebra.

• Comfortable with probability and statistics for data analysis or model evaluation.

Job Details:

Experience: 2-3 years of AI/ML or software development experience, including 1+ year in deep learning or GenAI projects.

Job Type: Hybrid

Job Time: 3pm to 9pm from office and 11pm to 1 am from home

Location: DHA Phase 6 Lahore

About Us:

TEKHQS is a global technology solutions provider headquartered in Lake Forest, California, with an offshore team of 300+ experts based in Pakistan. We specialize in Web 2.0 (Web & Mobile App Development), Web 3.0 (Blockchain & Crypto Platform Development), AI/ML Solutions, and ERP services as a certified partner of SAP S/4HANA, Oracle NetSuite, and Microsoft Dynamics 365 Business Central. Our expertise includes implementation, training, customization, integration, support, IT staff augmentation, and certified ERP consultancy.