As much as I’m a iPad-primary user. It took me quite a long time to determine this was the right fit for me, but you have to research on what application you will need for your studies. I understand you mention note taking and diagrams will be the bulk of your use cases, but there might be some applications you didn’t consider.
I’m sure majority will tell you to get the MBA, that’s the safe choice. But you can get by with the iPad… you have to research to see if the applications you need are on the platform. Whether it be web based application or found on the App Store.
This was the problem I ended up having--too many backflips to get iPadOS multitasking to behave in the way I expected it to behave. At first blush, iPad multitasking
looks a lot like dock-based multitasking and Spaces on macOS, where your full screen apps or app pairings live in their own Spaces, and you can put anything else you want in any manually created Spaces, and a quick swipe from the trackpad or mouse will allow you to flip through all your main work areas.
How iPad multitasking
acts though is completely different. Spaces is baked into macOS. Apps that work with macOS pretty much have to work in Spaces too. Apps that work with iPadOS are still not required (so far) to support multitasking, multiwindow, drag and drop, etc. Heck, some of them don't even support the iPad at all yet and have to run as iPhone apps, or don't allow you to rotate the iPad into landscape even there is a real iPad app.
My mistake was thinking the 12.9" iPad Pro would take me the rest of the way there. The 2020 model I had was wonderful, but due to its size, I started expecting even more laptop functionality out of it than I did my smaller iPads just due to the bigger workspace. That was a plus for drawing, which I don't do, but for laptop type tasks and even entertainment tasks, it ended up really magnifying some issues that didn't bother me as much on my smaller iPads.
I'm back on Mac now because the multitasking just sticks with me better. I always know where everything is, the gestures come more naturally to me, and all the iPad apps I'm missing from my old iPad Pro run on my M1 machine, so I'm covered.