Code Generation using GPT-J

About DeepGenX

CodeGenX is a Code Generation system powered by Artificial Intelligence! It is delivered to you in the form of a Visual Studio Code Extension and is Free and Open-source!

Languages Supported: CodeGenX currently only supports Python. We are planning to add additional languages in future releases.

Modules Trained On: CodeGenX was trained on Python code which covers many of its common uses. Some libraries which CodeGenX is specifically trained on are:

  • Tensorflow
  • Pytorch
  • Scikit-Learn
  • Pandas
  • NumPy
  • OpenCV
  • Django
  • Flask
  • PyGame

How CodeGenX Works: At the core of CodeGenX lies a large neural network called GPT-J. GPT-J is a 6 billion parameter transformer model which was trained on hundreds of gigabytes of text from the internet. We fine-tuned this model on a dataset of open-source python code. This fine-tuned model can now be used to generate code when given an input with the right instructions.

DeepGenX screenshots