Hello aspirant,
You have not mentioned the NIT or IIT name in which you are getting Computer Science Engineering branch and Mechanical Engineering/ Electrical Engineering branch respectively. It is difficult to say because some NITs are better than some IITs. Some IITs are better than any other NIT.
If you got Computer Science Engineering branch in NIT which is good and Mechanical Engineering or Electrical Engineering in IIT which is average or recently formed then go for Computer Science Engineering branch in NIT. If you have got Mechanical Engineering or Electrical Engineering branch in top IITs then go for it.
You can predict Colleges you can get for your JEE Mains rank using below mentioned link.
https://engineering.careers360.com/jee-main-college-predictor?utm_source=qna&utm_medium=jee_cp
May this information was helpful to you.
All The Best!!
Regular exam updates, QnA, Predictors, College Applications & E-books now on your Mobile