Dask: Getting the Row which has the max value in groups using groupby

The same problem can be solved in Pandas using transform as explained here With dask the only working solution I found use merge. And I was wondering if there are other ways to achieve it.