Pakistani officials are concerned about security in neighbouring Afghanistan, particularly about militant fighters from a separate Pakistani Taliban group crossing from Afghanistan and launching lethal attacks on its territory.
In the past few days, a suicide bombing claimed by an Afghan offshoot of ISIS outside Kabul airport killed more than 100 people, including 13 US troops.
A rocket attack on the airport followed, and on Sunday militant gunfire from across the border in Afghanistan killed two Pakistani soldiers.
"The next two to three months are critical," a senior Pakistani official told Reuters, adding that Islamabad feared a rise in militant attacks along the Afghan-Pakistan border, as the Taliban tried to fill a vacuum left by the collapse of Afghan forces and the western-backed administration.
"We (the international community) have to assist the Taliban in reorganising their army in order for them to control their territory," the official said, referring to the threat posed by resurgent rival militant groups, including Islamic State.
US officials have repeatedly accused Pakistan of supporting the Afghan Taliban.
Islamabad, one of the few capitals to recognise the Taliban government that was toppled in 2001, denies the charge.
Pakistan's government has said its influence over the movement has waned, particularly since the Taliban grew in confidence once Washington announced the date for the complete withdrawal of US and other foreign troops.
The official, who has direct knowledge of the country's security decisions, said Pakistan planned to send security and intelligence officials, possibly even the head of the powerful Inter-Services Intelligence (ISI) agency, to Kabul to help the Taliban reorganise the Afghan military.
Although recognition of a new Taliban government was not immediately on the table, the official said the world should not abandon Afghanistan.
"Whether we recognise the Taliban government or not, stability in Afghanistan is very important."
The official said ISIS-K, a loosely-affiliated offshoot of ISIS in Syria and Iraq, was actively looking to launch attacks and recruit new fighters.
Left unhindered, it would almost certainly grow from relatively small numbers currently.
The US recently launched two drone strikes targeting ISIS-K militants, including one in Kabul and one near the eastern border with Pakistan.
The strikes followed a pledge by President Joe Biden that the US would hunt down the militants behind the recent suicide bombing.
The Taliban criticised the strikes as a "clear attack on Afghan territory".
Pakistan, whose armed forces also possess unmanned drones, as well as conventional aircraft, will avoid intervening directly in Afghanistan if at all possible, the official said.
The Afghan Taliban have reassured their neighbour they will not allow their territory to be used by anyone planning attacks on Pakistan or any other country.
But Islamabad expected the Afghan Taliban to hand over militants planning attacks against Pakistan, the official added, or at least force them from their mutual border, where Pakistani troops have been on high alert in recent weeks.