En general, los lags del teclado suelen ser debidos a alguna aplicación que no tiene que ver pero que está consumiendo mucha memoria.
Yo tengo la 2.3.6 (KPQ con kernel de BAM) y el teclado me va de maravilla.
¿El teléfono ya te vino con la 2.3.6. o actualizaste desde una versión anterior?
Te lo digo porque si el teclado te funciona bien durantes unos minutos y luego se va quedando cada vez más lento, sospecho de alguna aplicación que se queda colgada o de alguna actualización de ROM que haya quedado mal...