Minor changes to dataset tutorial + adding explanation of InMemoryFederatedData.
Minor changes to dataset tutorial + adding explanation of InMemoryFederatedData.

PiperOrigin-RevId: 380809630

google/fedjax,google/fedjax