memo

import sys, platform
from memo import memo
import jax
import jax.numpy as np
from enum import IntEnum

print(f'Python {sys.version}')
print(f'Platform: {platform.platform()}')
print(f'Architecture: {platform.architecture()[0]}')
print(f'Machine: {platform.machine()}')
print(jax.__version__)

from memo import memo
import jax
import jax.numpy as np
from enum import IntEnum

print(jax.__version__)