def METHOD_NAME(path_to_model_files):
    import neomx  # noqa: F401
    ctx = mx.cpu()
    sym, arg_params, aux_params = mx.model.load_checkpoint(
        os.path.join(path_to_model_files, "compiled"), 0
    )
    mod = mx.mod.Module(symbol=sym, context=ctx, label_names=None)
    mod.bind(
        for_training=False, data_shapes=[("data", (1, 1, 28, 28))], label_shapes=mod._label_shapes
    )
    mod.set_params(arg_params, aux_params, allow_missing=True)
    return mod