Python - Sopa de letras - AIMA Python

Hola soy nuevo con Python, necesito un codigo que utilize la libreria AIMA, para resolver una sopa de letras.
El ejemplo que tengo es el siguiente.

Input:
w h x (its is the letter soup)
j e f
j k i

ax ( these are the words to look for)
ei
erh
ef

2 (It is the numbers of words to look for)

Ouput:

ef
ei