The deal includes the option for the Serie A side to purchase the 23-year-old permanently for £35.56 million, according to Sky in Italy (h/t Sky Sports).
Bakayoko will have a medical Monday before completing his move to the San Siro, per the report.
The midfielder only arrived at Chelsea last summer from Monaco after playing a key role in helping the Ligue 1 side win the French title.
However, he has struggled to live up to his £40 million price tag at Stamford Bridge and found it tough adapting to the Premier League.
His chances of regular football this season look even slimmer after Chelsea moved to strengthen their midfield. Jorginho has arrived from Napoli and Mateo Kovacic from Real Madrid.
The Blues also have Ross Barkley, Danny Drinkwater and Cesc Fabregas in the squad, which means competition for places is fierce.
Bakayoko did feature in pre-season but did not look too impressive, according to Mark Worrall at ESPN FC:
Half-time. Fair to say that, the Drinkwater / Bakayoko midfield combination has stilted Chelsea & affected the opportunities of those around them.— Mark Worrall (@gate17marco) August 7, 2018
The Premier League transfer window has closed, but it remains open throughout Europe, and the Blues can still sell and loan players.
New manager Maurizio Sarri has said Bakayoko is one player who could leave, per ESPN FC's Liam Twomey:
Sarri on his squad: "Now I think the squad is too large, so some players have to go on loan but I don’t know who. I prefer to have 23, 24 players & three, four very young players." Asked if Bakayoko could be one: "Maybe." #cfc— Liam Twomey (@liam_twomey) August 10, 2018
Chelsea have already agreed to loan out fringe players such as Kurt Zouma and Michy Batshuayi as they trim their squad.
Bakayoko looks to be next on the list. A departure from Stamford Bridge looks a good option for a player who is still only 23 but seems to have lost his way in London.
Football analyst Matteo Bonetti explained how he might be used at AC Milan:
In Milan's 4-3-3, Bakayoko would almost certainly be used as a mezz'ala - and I can't imagine him playing ahead of Kessie or Bonaventura. Nothing wrong with adding depth, though. Hopefully Tiemoue can find his confidence and regain his Monaco form in a new league.— Matteo Bonetti (@TheCalcioGuy) August 12, 2018
At his best, Bakayoko is a dynamic and aggressive midfielder capable of taking control of matches. He will be hoping a move to AC Milan can help him rediscover the form he showed at Monaco and get his career back on track.