Machine Learning Engineer - LLM/VLM

Job Description

About PureCode AI : PureCode is a front-end developer tool where engineers can use text to describe and generate or customize software user interfaces - (and soon entire projects). Our goal is to build a must use developer tool for front-end engineers to build web software 100x faster! We are headquartered in Austin, TX, USA with engineering offices in Hyderabad, India. This position is for exclusively in office work at our Q City Office in Hyderabad. Responsibilities : - Designing and implementing advanced state-of-the-art AI models, specializing in LLMs, VLMs and MLLMs. - Develop ML models, fine tune and work with stakeholders in finalising champion models. Perform scientific evaluation of NLP/LLM models, come up with new techniques for model validation, evaluation, trust and safety. - Understand how customer business needs are mapped to AI/ML problem and solution involving Algorithms/Models. - Support the process of Translating business problems into ML problems and create ML solutions to produce desired customer business outcomes. - Develop MLOps (Machine Learning Operations) workflows for data preparation, deployment, monitoring and retraining. Create and own cloud native API to deploy ML Models - Craft data warehousing strategy alongwith instrumenting ETL pipelines for maintaining quality data to be used for Model Training. - Design and implement A/B experiments - user segmentation, user classification, and tooling to support A/B experiments. Qualifications : - 3+ years of working experience developing, deploying, tracking and orchestrating scalable ML/AI solutions. - Familiarity with cloud platforms and services such as AWS, Azure, or GCP for deploying and scaling AI solutions. - Knowledge of basic ML stack - Pytorch, tensorflow, sklearn, numpy, pandas, etc. - Familiarity with experiment tracking tools like Weights & Biases, TensorBoard, or ClearML. - Hands on experience with MLOps tools like Mlflow, CometML, Docker, etc. - Hand on experience with workflow orchestration tools like Airflow, Prefect, or Databricks, etc. - Knowledge of API frameworks Django, Flask etc. - Knowledge of web development tools is a plus. - Knowledge of LLMops, langchain framework is a plus