@dom-miketa same here, I’ve also worked with TensorFlow and PyTorch for segmentation. As JAX is getting popular nowadays it would be amazing to learn it for segmentation tasks.
Yes, you’re right indeed I’ve read that paper but what confused me is if they’ve actually been able to reproduce the result because not all datasets are publicly available. Apart from that, TransUNet can be used for any medical image data. So, we can take advantage of it in either way.
Btw I live in the Bangladesh Standard Time (GMT+6). Looking forward to working with you.