AI Research Engineer

Job Description

Role : AI Research Engineer - AI Pair Programming About Us : We are developing a cutting-edge AI pair programming extension for VS Code, aimed at revolutionising the way developers write code. Our mission is to enhance developer productivity and code quality through advanced AI assistance. We're seeking an experienced AI Research Engineer to join our innovative team at PureCode AI and help push the boundaries of AI in software development. Position Overview : We are looking for an AI Research Engineer with a strong background in Large Language Models (LLMs) and their applications in code generation and understanding. The ideal candidate will contribute to our research efforts in contextual retrievals, rerankers, and vector databases, with a focus on optimising developer workflows. Key Responsibilities : - Conduct advanced research in AI and machine learning, specifically focusing on LLMs for code generation and comprehension - Implement or optimise state-of-the-art AI models to enhance our AI pair programming capabilities - Develop and refine algorithms for contextual code retrieval, reranking, and efficient use of vector databases - Collaborate with the development team to integrate research findings into our VS Code extension - Contribute to the improvement of developer workflow optimisation through AI-assisted coding - Evaluate and fine-tune models to ensure high-quality code suggestions and analysis - Stay abreast of the latest advancements in AI research and propose innovative applications for our product Qualifications : - Bachelor's or Master's degree in Computer Science, Artificial Intelligence, or a related field - Minimum of 3 years of experience in AI engineering and research, with a focus on NLP and LLMs - Strong background in machine learning, deep learning, and natural language processing - Expertise in implementing and deploying AI models for code analysis and generation - Experience with contextual retrieval systems, rerankers, and vector databases - Proficiency in Python and familiarity with machine learning frameworks such as PyTorch or TensorFlow - Solid understanding of API development practices and tools, particularly in the context of Generative AI. - Familiarity with VS Code extension development is a plus Required Skills : - Strong problem-solving skills and ability to translate research into practical applications - Excellent programming skills and ability to write clean, efficient code - Good understanding of software engineering principles and best practices - Ability to work effectively in a collaborative team environment - Strong communication skills to explain complex AI concepts to both technical and non-technical stakeholders - Self-motivated with the ability to work independently on research projects Preferred Experience : - Previous work on AI-assisted coding tools or IDE plugins - Contributions to open-source AI or developer tool projects - Experience with large-scale machine learning model deployment and optimisation - Familiarity with MLOps practices and tools What We Offer : - Opportunity to work on cutting-edge AI technology that directly impacts developers worldwide - Collaborative and innovative work environment - Competitive salary and benefits package - Professional development opportunities and support for attending relevant conferences - Chance to publish and present research findings in academic and industry forums Location- Delhi NCR, Bangalore, Chennai, Pune, Kolkata, Ahmedabad, Mumbai, Hyderabad