let memoize f =
  let table = Hashtbl.Poly.create () in
  fun x ->
  match Hashtbl.find table x with
  | Some y -> y
  | None ->
     let y = f x in
     Hashtbl.add_exn table ~key:x ~data:y;
     y