Add support for passing keyword arguments to tfds.load

[lgeiger]

tfds.load has gained some useful options in recent versions of tfds so it would be great to be able to pass them in.

In particular supporting read_config which has been introduced in [email protected] would be quite useful to support.