Further optimizations to custom GB models