At the 2017 World Aquatics Championships in the semifinal of the men's 100 metre butterfly, Milák set a new junior world record and a new national record (50.77). In the final he won the silver medal, and set a new junior world record and a new national record (50.62).