Skip to main content

A try to unify the deep learning frameworks under one api.

Project description


An attepmt to make a keras like framework for pytorch and tensorflow.

Read the Documentation.

What is Babilim?

Babilim is a Deep Learning Framework designed for ease of use like Keras. The API is designed to be intuitive and easy while allowing for complex production or research models. On top of that babilim runs on top of Tensorflow 2 or Pytorch, whichever you prefer. Seamless integration with TF2 and Pytorch was a top priority.

Babilim is designed for:

  • Intuitive usage,
  • a unified development experience across pytorch and tf2,
  • flexibility for research and robustness for production.


Follow the official instructions here to install pytorch or here to install tensorflow 2.

# Stable Version
pip install babilim

# Bleeding Edge
pip install git+

Selecting your Backend

Since babilim is designed for multiple backends (tensorflow 2 and pytorch), it gives you the choice which should be used. When your company/university uses one of the two frameworks you are fine to use or you can follow your personal preference for private projects.

import babilim
# or

Design Principles

Everything is attributed to one of three parts: Data, Model or Training. Some parts which are considered core functionality that is shared among them is in the core package.

  • Data is concerned about loading and preprocessing the data for training, evaluation and deployment.
  • Model is concerned with implementing the model. Everything required for the forward pass of the model is here.
  • Training contains all required for training a model on data. This includes loss, metrics, optimizers and trainers.

Tutorials & Examples

Starting with tutorials is usually easiest. The tutorials do not focus on the shortest possible solution, but actually an overkill solution that shows as much as you would need to know for solving your own problem.


Currently there are no guidelines on how to contribute, so the best thing you can do is open up an issue and get in contact that way. In the issue we can discuss how you can implement your new feature or how to fix that nasty bug.

Why called babilim?

TL;DR Reference to Tower of Babel.

The Tower of Babel narrative in Genesis 11:1-9 is a myth about mankind building a tower that could reach into the heavens. However, the attempt gets set back and fails because they started speaking different languages and were no longer able to understand each other.

Luckily for AI development there are only two major frameworks which share nearly all market share. Babilim is an attempt to unite the two to talk in a language compatible with both.

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

babilim-20201011.tar.gz (57.3 kB view hashes)

Uploaded source

Built Distribution

babilim-20201011-py3-none-any.whl (85.2 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page