Yes, you can be a TA for both courses as long as the schedule does not overlap with each other and there are no visa restrictions.