So, I pulled this out of another post on here (330d tuning options). This chip-tuning option is out of the UK, but apparently controls both fuel and boost (unlike JBD, which just does fuel). Looks to be semi-programmable too.
I checked out the site, and guess what? They have an option for the US-spec 335d. The module costs about $420 USD, with shipping included.