Give Pop_OS a try. It works well with optimus setups out of the box. You can install cuda, and tensorflow in a single apt command. It's a nice option for ML/DL work.