The AI assistant for data scientists & teams
Derive insights from your data faster.


Cogram uses artificial intelligence to give you code suggestions in your Jupyter Notebook.

Loved by data scientists

Cogram—a Codex-powered tool that stops you from needing to constantly look up data science code (main reason I lose patience with data science myself!), letting you "focus on the problems that matter".

Greg Brockman

This demo looks quite compelling, and I like the low-key UX approach taken - type prompt in plain comments...It will be interesting to explore how they can support/automate the more routine tasks and allow us to use our brain to actually better think about questions & data...

Fernando PĂ©rez

...a smart and well thought out symbiosis between man and machine might be happening in the context of code writing…

Lucas Soares
Machine Learning Engineer

Cogram: A Coding Assistant for Data Science and Machine Learning by @LucasEnkrateia

Towards Data Science

Why use Cogram?

Intelligent code suggestions

Cogram helps you write great code, fast, by making code suggestions from comments or existing code. Forget Googling "How to's" or Stack Overflow.

Learns from your code

Cogram is context aware, and reuses code you've already written. The more code you write with Cogram, the better Cogram’s code suggestions get.

Knows your team's codebase

For teams, Cogram can learn from your code repository to make use of existing code, such as utility functions and coding style. Get up to speed faster on a new codebase, and onboard new team members faster.

Use cases

Cogram works with Python and Julia in Jupyter Notebook, and SQL - some of the most common languages in data science.

Routine code

Don’t waste time on routine code. Open and save files, work with dictionaries, get files in a directory or manipulate data structures -- Cogram has you covered, for all common Python libraries.

Data analysis and Machine learning

Analyse your data with Cogram and your favorite libraries, such as Pandas, Scipy or TensorFlow. Skip looking up documentation and stay in flow.


Create beautiful visualisations, fast -- and customise them in plain language. Cogram knows libraries like Matplotlib, Plotly or Seaborn.

Database queries

Query databases with plain language and let Cogram generate the SQL code.

Frequently asked questions

What is Cogram?

Cogram is a coding assistant for data scientists that makes coding faster and easier. Cogram can write code from plain-language instructions or complete entire blocks of code. Cogram makes use of the Codex language model created by OpenAI.

How does Cogram work?

Cogram processes the context the user provides in the form of code or instructions in plain language, and then uses the OpenAI Codex language model to generate matching source code. The source code is returned to the user in their Notebook or the Cogram SQL app.


Cogram integrates with Jupyter Notebook, supporting the Python programming language. Cogram also supports SQL generation from plain language in the standalone Cogram SQL app. If you have integrations in mind, we’d love to hear from you at


The Codex language model used by Cogram was evaluated on a test-set of coding problems specified in plain language. The model generated the correct code for 30% of problems given a single attempt and for up to 70% of problems with a sampling strategy [Chen et al, ​​2021, Evaluating Large Language Models Trained on Code, arXiv:2107.03374].

How to best use Cogram

Cogram works especially well when you give it context: in an empty Jupyter Notebook, start out by writing a few lines of code manually before beginning to use Cogram. It also helps to split your workflow into smaller subproblems, and using meaningful names for functions, classes and variables.


In Jupyter Notebook, Cogram uses the content of your notebook above the current cursor position as context. In the Cogram SQL app, you can provide the database schema and an instruction of what you’d like your query to achieve as context.

Training set

Cogram makes use of the Codex language model developed by OpenAI. Codex was trained on large amounts of publicly available source code and natural language. An overview of the Codex model is available at here. A technical description is provided by Chen et al, ​​2021, Evaluating Large Language Models Trained on Code, arXiv:2107.03374.


In rare cases, Cogram can recite verbatim from the corpus of publicly available code and natural language that OpenAI’s Codex model was trained on. Typically, this happens when no or very little context is provided to Cogram. The Codex training set includes publicly available code repositories subject to licensing, such as the GNU General Public License (GPL). As a result, in the rare cases that Cogram recites verbatim from the training corpus, generated code may also be subject to licensing requirements. An investigation of verbatim recitation of the Codex model, in the context of the GitHub Copilot tool, is available here. Verbatim recitation was found in less than 0.1% of test cases. Code generated by Cogram should be vetted, just like code written by a human.


We use telemetry to improve Cogram. All data is transmitted and stored securely, anonymised, and never shared with other users or other companies. We collect information such as the size of the context, how many suggestions are requested, what settings are used, and whether or not a suggestion is accepted. At no point do we inspect, store, or process, the textual content of the context.


The context that the user provides is processed by Cogram and transmitted anonymously to the Codex API. The Codex API is provided by OpenAI and OpenAI may use the context to improve Codex. Details of how OpenAI uses the context are available here.

Can GitHub Copilot introduce insecure code in its suggestions?

There’s a lot of public code in the world with insecure coding patterns, bugs, or references to outdated APIs or idioms. When GitHub Copilot synthesizes code suggestions based on this data, it can also synthesize code that contains these undesirable patterns. This is something we care a lot about at GitHub, and in recent years we’ve provided tools such as Actions, Dependabot, and CodeQL to open source projects to help improve code quality. Similarly, as GitHub Copilot improves, we will work to exclude insecure or low-quality code from the training set. Of course, you should always use GitHub Copilot together with testing practices and security tools, as well as your own judgment.

Does GitHub Copilot produce offensive outputs?

The technical preview includes filters to block offensive words and avoid synthesizing suggestions in sensitive contexts. Due to the pre-release nature of the underlying technology, GitHub Copilot may sometimes produce undesired outputs, including biased, discriminatory, abusive, or offensive outputs. If you see offensive outputs, please report them directly to, so that we can improve our safeguards. GitHub takes this challenge very seriously and we are committed to addressing it with GitHub Copilot.

How will advanced code generation tools like GitHub Copilot affect developer jobs?

Bringing in more intelligent systems has the potential to bring enormous change to the developer experience. We expect this technology will enable existing engineers to be more productive, reducing manual tasks and helping them focus on interesting work. We also believe that GitHub Copilot has the potential to lower barriers to entry, enabling more people to explore software development and join the next generation of developers.

How is the data that GitHub Copilot collects used?

In order to generate suggestions, GitHub Copilot transmits part of the file you are editing to the service. This context is used to synthesize suggestions for you. GitHub Copilot also records whether the suggestions are accepted or rejected. This telemetry is used to improve future versions of the AI system, so that GitHub Copilot can make better suggestions for all users in the future. In the future we will give users the option to control how their telemetry is used. More information about our use of telemetry can be found here.

Is the transmitted data secure?

All data is transmitted and stored securely. Access to the telemetry is strictly limited to individuals on a need-to-know basis. Inspection of the gathered source code will be predominantly automatic, and when humans read it, it is specifically with the aim of improving the model or detecting abuse.

Will my private code be shared with other users?

No. We use telemetry data, including information about which suggestions users accept or reject, to improve the model. We do not reference your private code when generating code for other users.

Privacy & Data Protection

Please see the GitHub Copilot telemetry terms and About GitHub Copilot Telemetry. More information on how GitHub processes and uses personal data is available in our Privacy Statement.

Why is the technical preview restricted, and not available to everyone?

GitHub Copilot requires state-of-the-art AI hardware. During the technical preview, we are offering GitHub Copilot to a limited number of testers for free. When we launch a commercial product, we will make it available as broadly as possible.

Will there be a paid version?

If the technical preview is successful, our plan is to build a commercial version of GitHub Copilot in the future. We want to use the preview to learn how people use GitHub Copilot and what it takes to operate it at scale.

What development environments are supported?

We currently support Visual Studio Code, Neovim, and JetBrains IDEs like PyCharm and IntelliJ IDEA.

Sign up to our Newsletter

Stay up to date with the latest news from Cogram, and get regular product updates.
Thank you - we'll be in touch!
Something went wrong while submitting the form.