Machine Learning Engineer
Canada
Contributions:• Fine-tuned pre-trained models (open source text-to-image and LLMs) for various applications using Hugging Face libraries (Diffusers, PEFT, Transformers) and cloud computing instances (AWS EC2, GCP Compute Engine). • Deployed models to cloud services (GCP, AWS) and developed REST APIs (FastAPI). I also provided internal demos (using the Gradio Python library) for evaluation by the team, facilitating feedback and iterative improvement.• Participated in designing and developing novel applications of machine learning algorithms for new products. A color palette generator (a generative model using a GAN) and a color extraction tool from images(k-means). • Participated in data acquisition (web scraping, labeling) and data processing (cleaning, filtering) activities to develop new models.• Developed a retrieval augmented generation system (RAG) for better search functionality using OpenAI's GPT-4 and embeddings model. • Collaborated with cross-functional teams to define project requirements and deliverables.• Conducted research to stay up-to-date on the latest AI advancements and actively shared insights with the team to identify and explore creative solutions relevant to the startup.