# for Pytorch
import torch
import torch.nn as nn
import torchvision
import torchvision.transforms as transforms

# for tensorflow
import tensorflow as tf
from tensorflow import keras

# other usefull library
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline