<Button onClick={handleGoToPage1}>Go to page 1</Button>
<Box sx={{ height: 400, width: '100%' }}>
  <DataGrid
    {...data}
    loading={loading}
    apiRef={apiRef}
    pagination
    paginationModel={paginationModel}
    onPaginationModelChange={setPaginationModel}
  />
</Box>