Jasprit Bumrah is set to return after an injury lay-off that has lasted nearly three months.