from absl import app
from absl import flags
from absl import logging

import jax as jax
import jax.numpy as np

from jax.api import jit

import sys
import os
import numpy as onp
import matplotlib.pyplot as plt


def timeLoop():
  return 7
