AI/ML Engineer- Python

We are seeking a passionate AI/ML Engineer to join our team in building the core AI-driven functionality an intelligent visual data encryption system.

The role involves designing, training, and deploying AI models (e.g., CLIP, DCGANs, Decision Trees), integrating them into a secure backend, and operationalizing the solution via AWS cloud services and Python-based APIs.

Key Responsibilities:

• AI/ML Development
• Design and train deep learning models for image classification and sensitivity tagging using CLIP, DCGANs, and Decision Trees.
• Build synthetic datasets using DCGANs for balancing.
• Fine-tune pre-trained models for customized encryption logic.
• Implement explainable classification logic for model outputs.
• Validate model performance using custom metrics and datasets.
• API Development
• Design and develop Python RESTful APIs using FastAPI or Flask for:
• Image upload and classification
• Model inference endpoints
• Encryption trigger calls
• Integrate APIs with AWS Lambda and Amazon API Gateway.
• AWS Integration
• Deploy and manage AI models on Amazon SageMaker for training and real-time inference.
• Use AWS Lambda for serverless backend compute.
• Store encrypted image data on Amazon S3 and metadata on Amazon RDS (PostgreSQL).
• Use AWS Cognito for secure user authentication and KMS for key management.
• Monitor job status via CloudWatch and enable secure, scalable API access.
• Required Skills & Experience:
• Must-Have
• 3–5 years of experience in AI/ML (especially vision-based systems).
• Strong experience with PyTorch or TensorFlow for model development.
• Proficient in Python with experience building RESTful APIs.
• Hands-on experience with Amazon SageMaker, Lambda, API Gateway, and S3.
• Knowledge of OpenSSL/PyCryptodome or basic cryptographic concepts.
• Understanding of model deployment, serialization, and performance tuning.
• Nice-to-Have
• Experience with CLIP model fine-tuning.
• Familiarity with Docker, GitHub Actions, or CI/CD pipelines.
• Experience in data classification under compliance regimes (e.g., GDPR, HIPAA).
• Familiarity with multi-tenant SaaS design patterns.

Tools & Technologies:

Python, PyTorch, TensorFlow

FastAPI, Flask

AWS: SageMaker, Lambda, S3, RDS, Cognito, API Gateway, KMS

Git, Docker, Postgres, OpenCV, OpenSSL