Chanlaw

# %%
from typing import Tuple

import einops
import numpy as np
import pandas as pd  # type: ignore
import seaborn as sns  # type: ignore
import torch

n_steps = 10000