microTorch: A Software Package for Fast and Flexible Self-Supervised Diffusion MRI Model Fitting

The microTorch software package allows for the fast and flexible fitting of a wide range of multi-compartment microstructure models to diffusion magnetic resonance imaging (dMRI) data. It employs a self-supervised learning approach, which enables the speed gains of deep learning whilst minimising estimation bias induced by the training data distribution. Its modular architecture enables users to readily evaluate different combinations of compartment models, apply established multi-compartment models, and interchange neural network architectures without modifying the underlying framework. The package supports custom acquisition schemes and provides configurable training settings, allowing it to be tailored to specific datasets. microTorch is built using PyTorch and is freely available at https://github.com/snigdha-sen/microtorch.