I don't know if it's OK to make those jokes, but it's infinitely preferable to say it out loud explicitly and deal with it with humor than to just make fun of them for being weird and acting different like on big bang theory, where sheldon is clearly modeled after someone on the spectrum but with few redeeming qualities.