
Generative AI extension for JupyterLab and Jupyter Notebooks
Jupyter AI is an open-source extension that brings generative AI capabilities directly into JupyterLab and Jupyter Notebooks. It provides a native chat interface, the %%ai magic command for reproducible AI workflows, and supports over 1000 LLMs from providers like OpenAI, Anthropic, Google, and more, including local models via Ollama and GPT4All.
Native sidebar chat UI in JupyterLab for conversational AI assistance with code, data analysis, and notebook tasks
Cell magic that turns any Jupyter notebook into a reproducible generative AI playground, works in JupyterLab, Colab, and VS Code
Out-of-the-box support for over 1000 LLMs from AI21, Anthropic, AWS, Cohere, Gemini, Hugging Face, MistralAI, NVIDIA, and OpenAI
Run AI models locally with GPT4All and Ollama for privacy-focused and offline usage on consumer-grade hardware
Multiple editable chat threads saved as Jupyter server documents with .chat extension for organized conversations
Generate entire notebooks from scratch using natural language prompts, including data preprocessing and model training code
AI assistant that can learn about your local data and provide more specific, relevant responses based on your project context
Use AI to quickly generate code for data preprocessing, exploratory analysis, and visualization directly within notebooks
Generate model training pipelines, feature engineering code, and evaluation scripts through conversational prompts
Automatically generate docstrings, comments, and explanations for existing notebook code using the chat interface
Students and educators can use AI assistance to understand concepts, debug code, and explore algorithms interactively
Migrated from LangChain to LiteLLM for faster startup times (reduced by 75%) and broader model compatibility
Run local models via Ollama or GPT4All for sensitive data analysis without sending data to external APIs

Free, open-source code editor from Microsoft