Artificial Intelligence Engineer

Job DescriptionAs an Artificial Intelligence Engineer at JANA Inc., you will play a pivotal role in developing AIsolutions that streamline and enhance the authoring of technical documents. You will work closelywith cross-functional teams to build innovative tools and applications.Key ResponsibilitiesDevelop and implement Machine Learning (ML), Large Language Models (LLMs), and Computer Vision solutions for tasks such as classification, prediction, object detection, OCR, image processing, text summarization, text generation, and information extraction.Design, build, and maintain APIs, including high-performance services with FastAPI, for internal and external consumption.Integrate and operate third-party AI services and SDKs (e.g., OpenAI, Google Gemini, Mistral, Ollama) and manage authentication, rate limits, and observability.Collaborate with technical writers to understand requirements and challenges in technical document authoring.Collect, preprocess, and annotate data to create high-quality datasets for training and evaluation; design and maintain databases to store and version data and model artifacts.Optimize, fine-tune, and evaluate ML, LLM, and Computer Vision models to improve accuracy, efficiency, and user experience.Design and implement Retrieval-Augmented Generation (RAG) architectures, including document ingestion, indexing, retrieval strategies, and grounding.Apply Reinforcement Learning from Human Feedback (RLHF) (and related techniques such as PPO/DPO) to improve LLM alignment and performance is a plus.Participate in code reviews, testing, and debugging to ensure robust, scalable, and maintainable AI services.Collaborate with the software engineering team to integrate AI capabilities into existing or new applications.Communicate technical findings, challenges, and progress to stakeholders.Stay current with advances in ML, LLMs, NLP, and Computer Vision, and apply state-of-the-art techniques to our toolchain.RequirementsBachelor's or Master's degree in Artificial Intelligence, Computer Science, Data Science, Machine Learning, Software Development, or a related field; a Ph.D. is a plus.2-3 years of verifiable, hands-on experience in AI projects or processes, demonstrating ongoing, practical involvement.Proven experience building ML models and/or LLM-based/NLP solutions, including Computer Vision pipelines (object detection, OCR) and model fine-tuning.Proficiency in Python and ML/LLM ecosystems (TensorFlow, PyTorch, Pandas, NumPy, SciPy).Practical experience designing and developing APIs (preferably with FastAPI) and integrating external AI services (OpenAI, Google Gemini, Mistral, Ollama).Working knowledge of LangChain and LangGraph for LLM application orchestration and tool/agent design.Solid understanding of ML, LLMs, NLP techniques, data cleaning/preprocessing, and feature engineering.Experience with RAG architectures training/evaluation workflows.Strong proficiency with Linux environments (shell, packaging, permissions, basic networking, and debugging).Strong problem-solving skills and ability to work in a collaborative team environment.Excellent communication skills.Knowledge of version control systems and software development best practices.Experience with cloud platforms (e.g., AWS, Azure, or GCP) is a plus.