Понял, что прямого API нет, но этого можно достичь, объединив два вызова.
partitions_for(topic) - Returns set of all known partitions for the topic.
send(topic, value=None, key=None, headers=None, partition=None, timestamp_ms=None) - Publish a message to a topic.