added backend support for storing td0 weights, updated ai