3D Diffusion lib