New mum Angelina Jolie is set to go back to work with her first film since giving birth to daughter Shiloh at the end of May.