ML Engineer – NLP and Large Language Models (LLM)

We expect you to bring the following experience and qualifications to this position.

1. A bachelor’s degree or higher in computer science, data science, artificial intelligence, or a related field
2. Previous experience in developing AI models for SQL generation from natural language queries.
3. Familiarity with SQL generation tools and techniques, including query templates and SQL builders, is essential.
4. Knowledge of databases and data management concepts is important for working with SQL databases.
5. Familiarity with NLP pre-trained models (e.g., ChatGPT, Langchain, StarCoder, SQLCoder).
6. Strong programming skills, with experience in languages like Python.
7. Familiarity with code analysis tools, static analysis, and code quality assessment.
8. Ability to work collaboratively with software development teams.
9. Effective communication skills for technical and non-technical stakeholders.
10. A proactive and creative approach to problem-solving.
11. Dedication to staying updated with emerging AI technologies.

1. Research and implement state-of-the-art techniques and frameworks for LLM training and inference, such as ChatGPT, Langchain, StarCoder, SQLCoder, etc.
2. Train, fine-tune, and evaluate LLMs on various datasets, benchmarks, and evaluation metrics, such as WikiSQL, Spider, HumanEval, etc.
3. Experiment with different prompt engineering, in-context learning, zero-shot learning, and other methods to improve the performance and robustness of LLMs for various tasks, such as natural language to SQL (NL2SQL), natural language to code (NL2Code), code analysis, code generation, etc.
4. Collaborate with cross-functional teams and stakeholders to understand the business requirements and use cases for LLM-based products
5. Communicate and document the technical details and results of the LLMs and products