TorchDiff 2.0.0__tar.gz

torchdiff-2.0.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Loghman Samani
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

torchdiff-2.0.0/PKG-INFO

@@ -0,0 +1,315 @@
+Metadata-Version: 2.4
+Name: TorchDiff
+Version: 2.0.0
+Summary: A PyTorch-based library for diffusion models
+Home-page: https://github.com/LoqmanSamani/TorchDiff
+Author: Loghman Samani
+Author-email: samaniloqman91@gmail.com
+License: MIT
+Project-URL: Homepage, https://loqmansamani.github.io/torchdiff
+Project-URL: Documentation, https://torchdiff.readthedocs.io
+Project-URL: Source, https://github.com/LoqmanSamani/TorchDiff
+Keywords: diffusion models,pytorch,machine learning,deep learning
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: lpips==0.1.4
+Requires-Dist: pytorch-fid==0.3.0
+Requires-Dist: torch==2.7.0
+Requires-Dist: torchvision==0.22.0
+Requires-Dist: tqdm==4.67.1
+Requires-Dist: transformers==4.51.3
+Provides-Extra: test
+Requires-Dist: pytest>=7.0.0; extra == "test"
+Requires-Dist: pytest-cov>=4.0.0; extra == "test"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: project-url
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# TorchDiff
+
+<div align="center">
+  <img src="imgs/logo_.png" alt="TorchDiff Logo" width="300"/>
+</div>
+
+<div align="center">
+
+[![License: MIT](https://img.shields.io/badge/license-MIT-red?style=plastic)](https://opensource.org/licenses/MIT)
+[![PyTorch](https://img.shields.io/badge/PyTorch-white?style=plastic&logo=pytorch&logoColor=red)](https://pytorch.org/)
+[![Version](https://img.shields.io/badge/version-2.0.0-blue?style=plastic)](https://pypi.org/project/torchdiff/)
+[![Python](https://img.shields.io/badge/python-3.8%2B-blue?style=plastic&logo=python&logoColor=white)](https://www.python.org/)
+[![Downloads](https://pepy.tech/badge/torchdiff)](https://pepy.tech/project/torchdiff)
+[![Stars](https://img.shields.io/github/stars/LoqmanSamani/TorchDiff?style=plastic&color=yellow)](https://github.com/LoqmanSamani/TorchDiff)
+[![Forks](https://img.shields.io/github/forks/LoqmanSamani/TorchDiff?style=plastic&color=orange)](https://github.com/LoqmanSamani/TorchDiff)
+[![Issues](https://img.shields.io/github/issues/LoqmanSamani/TorchDiff?style=plastic&color=red)](https://github.com/LoqmanSamani/TorchDiff/issues)
+
+</div>
+
+---
+
+## 🔎 Overview  
+
+**TorchDiff** is a PyTorch-based library for building and experimenting with diffusion models, inspired by leading research papers.  
+
+The **TorchDiff 2.0.0** release includes implementations of five major diffusion model families:  
+- **DDPM** (Denoising Diffusion Probabilistic Models)  
+- **DDIM** (Denoising Diffusion Implicit Models)  
+- **SDE-based Diffusion**  
+- **LDM** (Latent Diffusion Models)  
+- **UnCLIP** (the model powering OpenAI’s *DALL·E 2*)  
+
+These models support both **conditional** (e.g., text-to-image) and **unconditional** generation.  
+
+<div align="center">
+  <img src="imgs/mount.png" alt="Diffusion Model Process" width="1000"/>
+  <br>
+  <em>Image generated using Sora</em>
+  <br><br>
+</div>
+
+TorchDiff is designed with **modularity** in mind. Each model is broken down into reusable components:  
+- **Forward Diffusion**: Adds noise (e.g., `ForwardDDPM`).  
+- **Reverse Diffusion**: Removes noise to recover data (e.g., `ReverseDDPM`).  
+- **Variance Scheduler**: Controls noise schedules (e.g., `VarianceSchedulerDDPM`).  
+- **Training**: Full training pipelines (e.g., `TrainDDPM`).  
+- **Sampling**: Efficient inference and generation (e.g., `SampleDDPM`).  
+
+Additional utilities:  
+- **Noise Predictor**: A U-Net-like model with attention and time embeddings.  
+- **Text Encoder**: Transformer-based (e.g., BERT) for conditional generation.  
+- **Metrics**: Evaluation suite including MSE, PSNR, SSIM, FID, and LPIPS.  
+
+---
+
+## ⚡ Quick Start  
+
+Here’s a minimal working example to train and sample with **DDPM** on dummy data:  
+
+```python
+import torch
+import torch.nn as nn
+from torchvision import datasets, transforms
+from torch.utils.data import DataLoader
+
+from torchdiff.ddpm import VarianceSchedulerDDPM, ForwardDDPM, ReverseDDPM, TrainDDPM, SampleDDPM
+from torchdiff.utils import NoisePredictor
+
+# Dataset (CIFAR10 for demo)
+transform = transforms.Compose([
+    transforms.Resize(32),
+    transforms.ToTensor(),
+    transforms.Normalize((0.5,), (0.5,))
+])
+train_dataset = datasets.CIFAR10(root="./data", train=True, download=True, transform=transform)
+train_loader = DataLoader(train_dataset, batch_size=32, shuffle=True)
+
+# Model components
+noise_pred = NoisePredictor(in_channels=3)
+vs = VarianceSchedulerDDPM(num_steps=1000)
+fwd, rev = ForwardDDPM(vs), ReverseDDPM(vs)
+
+# Optimizer & loss
+optim = torch.optim.Adam(noise_pred.parameters(), lr=1e-4)
+loss_fn = nn.MSELoss()
+
+# Training
+trainer = TrainDDPM(
+    noise_predictor=noise_pred, forward_diffusion=fwd, reverse_diffusion=rev,
+    conditional_model=None, optimizer=optim, objective=loss_fn,
+    data_loader=train_loader, max_epochs=1, device="cpu"
+)
+trainer()
+
+# Sampling
+sampler = SampleDDPM(reverse_diffusion=rev, noise_predictor=noise_pred,
+                     image_shape=(32, 32), batch_size=4, in_channels=3, device="cpu")
+images = sampler()
+print("Generated images shape:", images.shape)
+```
+
+For detailed examples, check the [examples/](https://github.com/LoqmanSamani/TorchDiff/tree/systembiology/examples) directory.  
+
+---
+
+## 📚 Resources  
+- 🌐 [Project Website](https://loqmansamani.github.io/torchdiff/)  
+- 📖 [API Reference](https://torchdiff.readthedocs.io/en/latest/index.html)  
+
+---
+
+## ⚡ Installation  
+
+Install from **PyPI (recommended):**
+```bash
+pip install torchdiff
+```
+
+Or install from source for development:  
+```bash
+# Clone repository
+git clone https://github.com/LoqmanSamani/TorchDiff.git
+cd TorchDiff
+
+# Install dependencies
+pip install -r requirements.txt
+
+# Install package
+pip install .
+```
+
+> Requires **Python 3.8+**. For GPU acceleration, ensure PyTorch is installed with the correct CUDA version.  
+
+---
+
+## 🧩 Implemented Models  
+
+### 1. Denoising Diffusion Probabilistic Models (DDPM)  
+**Paper**: [Ho et al., 2020](https://arxiv.org/abs/2006.11239)  
+
+DDPMs learn to reverse a gradual noise-adding process to generate high-quality images. TorchDiff provides a modular implementation for both unconditional and conditional (text-guided) generation.  
+
+📓 [DDPM Example Notebook](https://github.com/LoqmanSamani/TorchDiff/blob/systembiology/examples/ddpm.ipynb)  
+
+---
+
+### 2. Denoising Diffusion Implicit Models (DDIM)  
+**Paper**: [Song et al., 2021](https://arxiv.org/abs/2010.02502)  
+
+DDIM accelerates sampling by reducing the number of denoising steps while maintaining image quality. TorchDiff supports both conditional and unconditional DDIM generation.  
+
+📓 [DDIM Example Notebook](https://github.com/LoqmanSamani/TorchDiff/blob/systembiology/examples/ddim.ipynb)  
+
+---
+
+### 3. Score-Based Generative Models via Stochastic Differential Equations (SDE)  
+**Paper**: [Song et al., 2021](https://arxiv.org/abs/2011.13456)  
+
+SDE-based models generalize diffusion via stochastic processes, supporting multiple formulations: **VE, VP, sub-VP**, and deterministic **ODE** variants. TorchDiff includes full training and sampling pipelines for both conditional and unconditional use cases.  
+
+📓 [SDE Example Notebook](https://github.com/LoqmanSamani/TorchDiff/blob/systembiology/examples/sde.ipynb)  
+
+---
+
+### 4. Latent Diffusion Models (LDM)  
+**Paper**: [Rombach et al., 2022](https://arxiv.org/abs/2112.10752)  
+
+LDMs operate in a compressed latent space using a VAE, enabling **efficient high-resolution image synthesis** with reduced computational cost. TorchDiff supports using DDPM, DDIM, or SDE as the diffusion backbone in latent space.  
+
+📓 [LDM Example Notebook](https://github.com/LoqmanSamani/TorchDiff/blob/systembiology/examples/ldm.ipynb)  
+
+---
+
+### 5. UnCLIP (Hierarchical Text-Conditional Image Generation with CLIP Latents)  
+**Paper**: [Ramesh et al., 2022](https://arxiv.org/abs/2204.06125)  
+
+UnCLIP, the architecture behind *DALL·E 2*, leverages **CLIP latents** to enable hierarchical text-to-image generation. It first maps text into CLIP’s multimodal embedding space, then performs diffusion-based generation in that space, followed by refinement in pixel space.  
+
+Training UnCLIP is significantly more complex than other diffusion families, and thus a minimal example is not shown here.  
+
+📓 [UnCLIP Example Notebook](https://github.com/LoqmanSamani/TorchDiff/blob/systembiology/examples/unclip.ipynb)  
+
+---
+
+## 🔐 License  
+Released under the [MIT License](https://github.com/LoqmanSamani/TorchDiff/blob/systembiology/LICENSE).  
+
+---
+
+## 🚧 Roadmap / Future Work  
+TorchDiff is under active development. Planned features include:  
+- 🧠 New diffusion variants and improved training algorithms.  
+- ⚡ Faster and more memory-efficient sampling.  
+- 🎯 Additional utilities to simplify experimentation.  
+
+---
+
+## 🤝 Contributing  
+Contributions are welcome!  
+
+- Open an [Issue](../../issues) to report bugs or request features.  
+- Submit a PR with improvements or new features.  
+
+Your feedback helps make TorchDiff better for the community.  
+
+
+---
+
+## 📖 Citation  
+
+If you use **TorchDiff** in your research or project, please cite the original papers and this repository.  
+
+### Core Diffusion Papers  
+
+```bibtex
+@article{ho2020denoising,
+  title={Denoising Diffusion Probabilistic Models},
+  author={Ho, Jonathan and Jain, Ajay and Abbeel, Pieter},
+  journal={Advances in Neural Information Processing Systems},
+  year={2020}
+}
+
+@article{song2021denoising,
+  title={Denoising Diffusion Implicit Models},
+  author={Song, Jiaming and Meng, Chenlin and Ermon, Stefano},
+  journal={International Conference on Learning Representations (ICLR)},
+  year={2021}
+}
+
+@article{song2021score,
+  title={Score-Based Generative Modeling through Stochastic Differential Equations},
+  author={Song, Yang and Sohl-Dickstein, Jascha and Kingma, Diederik P and Kumar, Abhishek and Ermon, Stefano and Poole, Ben},
+  journal={International Conference on Learning Representations (ICLR)},
+  year={2021}
+}
+
+@article{rombach2022high,
+  title={High-Resolution Image Synthesis with Latent Diffusion Models},
+  author={Rombach, Robin and Blattmann, Andreas and Lorenz, Dominik and Esser, Patrick and Ommer, Björn},
+  journal={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
+  year={2022}
+}
+
+@article{ramesh2022hierarchical,
+  title={Hierarchical Text-Conditional Image Generation with CLIP Latents},
+  author={Ramesh, Aditya and Pavlov, Mikhail and Goh, Gabriel and Gray, Scott and Voss, Chelsea and Radford, Alec and Chen, Mark and Sutskever, Ilya},
+  journal={arXiv preprint arXiv:2204.06125},
+  year={2022}
+}
+```
+
+### TorchDiff Repository  
+
+```bibtex
+@misc{torchdiff2025,
+  author = {Samani, Loghman},
+  title = {TorchDiff: A Modular Diffusion Modeling Library in PyTorch},
+  year = {2025},
+  publisher = {GitHub},
+  journal = {GitHub repository},
+  howpublished = {\url{https://github.com/LoqmanSamani/TorchDiff}},
+}
+```
+

torchdiff-2.0.0/README.md

@@ -0,0 +1,264 @@
+# TorchDiff
+
+<div align="center">
+  <img src="imgs/logo_.png" alt="TorchDiff Logo" width="300"/>
+</div>
+
+<div align="center">
+
+[![License: MIT](https://img.shields.io/badge/license-MIT-red?style=plastic)](https://opensource.org/licenses/MIT)
+[![PyTorch](https://img.shields.io/badge/PyTorch-white?style=plastic&logo=pytorch&logoColor=red)](https://pytorch.org/)
+[![Version](https://img.shields.io/badge/version-2.0.0-blue?style=plastic)](https://pypi.org/project/torchdiff/)
+[![Python](https://img.shields.io/badge/python-3.8%2B-blue?style=plastic&logo=python&logoColor=white)](https://www.python.org/)
+[![Downloads](https://pepy.tech/badge/torchdiff)](https://pepy.tech/project/torchdiff)
+[![Stars](https://img.shields.io/github/stars/LoqmanSamani/TorchDiff?style=plastic&color=yellow)](https://github.com/LoqmanSamani/TorchDiff)
+[![Forks](https://img.shields.io/github/forks/LoqmanSamani/TorchDiff?style=plastic&color=orange)](https://github.com/LoqmanSamani/TorchDiff)
+[![Issues](https://img.shields.io/github/issues/LoqmanSamani/TorchDiff?style=plastic&color=red)](https://github.com/LoqmanSamani/TorchDiff/issues)
+
+</div>
+
+---
+
+## 🔎 Overview  
+
+**TorchDiff** is a PyTorch-based library for building and experimenting with diffusion models, inspired by leading research papers.  
+
+The **TorchDiff 2.0.0** release includes implementations of five major diffusion model families:  
+- **DDPM** (Denoising Diffusion Probabilistic Models)  
+- **DDIM** (Denoising Diffusion Implicit Models)  
+- **SDE-based Diffusion**  
+- **LDM** (Latent Diffusion Models)  
+- **UnCLIP** (the model powering OpenAI’s *DALL·E 2*)  
+
+These models support both **conditional** (e.g., text-to-image) and **unconditional** generation.  
+
+<div align="center">
+  <img src="imgs/mount.png" alt="Diffusion Model Process" width="1000"/>
+  <br>
+  <em>Image generated using Sora</em>
+  <br><br>
+</div>
+
+TorchDiff is designed with **modularity** in mind. Each model is broken down into reusable components:  
+- **Forward Diffusion**: Adds noise (e.g., `ForwardDDPM`).  
+- **Reverse Diffusion**: Removes noise to recover data (e.g., `ReverseDDPM`).  
+- **Variance Scheduler**: Controls noise schedules (e.g., `VarianceSchedulerDDPM`).  
+- **Training**: Full training pipelines (e.g., `TrainDDPM`).  
+- **Sampling**: Efficient inference and generation (e.g., `SampleDDPM`).  
+
+Additional utilities:  
+- **Noise Predictor**: A U-Net-like model with attention and time embeddings.  
+- **Text Encoder**: Transformer-based (e.g., BERT) for conditional generation.  
+- **Metrics**: Evaluation suite including MSE, PSNR, SSIM, FID, and LPIPS.  
+
+---
+
+## ⚡ Quick Start  
+
+Here’s a minimal working example to train and sample with **DDPM** on dummy data:  
+
+```python
+import torch
+import torch.nn as nn
+from torchvision import datasets, transforms
+from torch.utils.data import DataLoader
+
+from torchdiff.ddpm import VarianceSchedulerDDPM, ForwardDDPM, ReverseDDPM, TrainDDPM, SampleDDPM
+from torchdiff.utils import NoisePredictor
+
+# Dataset (CIFAR10 for demo)
+transform = transforms.Compose([
+    transforms.Resize(32),
+    transforms.ToTensor(),
+    transforms.Normalize((0.5,), (0.5,))
+])
+train_dataset = datasets.CIFAR10(root="./data", train=True, download=True, transform=transform)
+train_loader = DataLoader(train_dataset, batch_size=32, shuffle=True)
+
+# Model components
+noise_pred = NoisePredictor(in_channels=3)
+vs = VarianceSchedulerDDPM(num_steps=1000)
+fwd, rev = ForwardDDPM(vs), ReverseDDPM(vs)
+
+# Optimizer & loss
+optim = torch.optim.Adam(noise_pred.parameters(), lr=1e-4)
+loss_fn = nn.MSELoss()
+
+# Training
+trainer = TrainDDPM(
+    noise_predictor=noise_pred, forward_diffusion=fwd, reverse_diffusion=rev,
+    conditional_model=None, optimizer=optim, objective=loss_fn,
+    data_loader=train_loader, max_epochs=1, device="cpu"
+)
+trainer()
+
+# Sampling
+sampler = SampleDDPM(reverse_diffusion=rev, noise_predictor=noise_pred,
+                     image_shape=(32, 32), batch_size=4, in_channels=3, device="cpu")
+images = sampler()
+print("Generated images shape:", images.shape)
+```
+
+For detailed examples, check the [examples/](https://github.com/LoqmanSamani/TorchDiff/tree/systembiology/examples) directory.  
+
+---
+
+## 📚 Resources  
+- 🌐 [Project Website](https://loqmansamani.github.io/torchdiff/)  
+- 📖 [API Reference](https://torchdiff.readthedocs.io/en/latest/index.html)  
+
+---
+
+## ⚡ Installation  
+
+Install from **PyPI (recommended):**
+```bash
+pip install torchdiff
+```
+
+Or install from source for development:  
+```bash
+# Clone repository
+git clone https://github.com/LoqmanSamani/TorchDiff.git
+cd TorchDiff
+
+# Install dependencies
+pip install -r requirements.txt
+
+# Install package
+pip install .
+```
+
+> Requires **Python 3.8+**. For GPU acceleration, ensure PyTorch is installed with the correct CUDA version.  
+
+---
+
+## 🧩 Implemented Models  
+
+### 1. Denoising Diffusion Probabilistic Models (DDPM)  
+**Paper**: [Ho et al., 2020](https://arxiv.org/abs/2006.11239)  
+
+DDPMs learn to reverse a gradual noise-adding process to generate high-quality images. TorchDiff provides a modular implementation for both unconditional and conditional (text-guided) generation.  
+
+📓 [DDPM Example Notebook](https://github.com/LoqmanSamani/TorchDiff/blob/systembiology/examples/ddpm.ipynb)  
+
+---
+
+### 2. Denoising Diffusion Implicit Models (DDIM)  
+**Paper**: [Song et al., 2021](https://arxiv.org/abs/2010.02502)  
+
+DDIM accelerates sampling by reducing the number of denoising steps while maintaining image quality. TorchDiff supports both conditional and unconditional DDIM generation.  
+
+📓 [DDIM Example Notebook](https://github.com/LoqmanSamani/TorchDiff/blob/systembiology/examples/ddim.ipynb)  
+
+---
+
+### 3. Score-Based Generative Models via Stochastic Differential Equations (SDE)  
+**Paper**: [Song et al., 2021](https://arxiv.org/abs/2011.13456)  
+
+SDE-based models generalize diffusion via stochastic processes, supporting multiple formulations: **VE, VP, sub-VP**, and deterministic **ODE** variants. TorchDiff includes full training and sampling pipelines for both conditional and unconditional use cases.  
+
+📓 [SDE Example Notebook](https://github.com/LoqmanSamani/TorchDiff/blob/systembiology/examples/sde.ipynb)  
+
+---
+
+### 4. Latent Diffusion Models (LDM)  
+**Paper**: [Rombach et al., 2022](https://arxiv.org/abs/2112.10752)  
+
+LDMs operate in a compressed latent space using a VAE, enabling **efficient high-resolution image synthesis** with reduced computational cost. TorchDiff supports using DDPM, DDIM, or SDE as the diffusion backbone in latent space.  
+
+📓 [LDM Example Notebook](https://github.com/LoqmanSamani/TorchDiff/blob/systembiology/examples/ldm.ipynb)  
+
+---
+
+### 5. UnCLIP (Hierarchical Text-Conditional Image Generation with CLIP Latents)  
+**Paper**: [Ramesh et al., 2022](https://arxiv.org/abs/2204.06125)  
+
+UnCLIP, the architecture behind *DALL·E 2*, leverages **CLIP latents** to enable hierarchical text-to-image generation. It first maps text into CLIP’s multimodal embedding space, then performs diffusion-based generation in that space, followed by refinement in pixel space.  
+
+Training UnCLIP is significantly more complex than other diffusion families, and thus a minimal example is not shown here.  
+
+📓 [UnCLIP Example Notebook](https://github.com/LoqmanSamani/TorchDiff/blob/systembiology/examples/unclip.ipynb)  
+
+---
+
+## 🔐 License  
+Released under the [MIT License](https://github.com/LoqmanSamani/TorchDiff/blob/systembiology/LICENSE).  
+
+---
+
+## 🚧 Roadmap / Future Work  
+TorchDiff is under active development. Planned features include:  
+- 🧠 New diffusion variants and improved training algorithms.  
+- ⚡ Faster and more memory-efficient sampling.  
+- 🎯 Additional utilities to simplify experimentation.  
+
+---
+
+## 🤝 Contributing  
+Contributions are welcome!  
+
+- Open an [Issue](../../issues) to report bugs or request features.  
+- Submit a PR with improvements or new features.  
+
+Your feedback helps make TorchDiff better for the community.  
+
+
+---
+
+## 📖 Citation  
+
+If you use **TorchDiff** in your research or project, please cite the original papers and this repository.  
+
+### Core Diffusion Papers  
+
+```bibtex
+@article{ho2020denoising,
+  title={Denoising Diffusion Probabilistic Models},
+  author={Ho, Jonathan and Jain, Ajay and Abbeel, Pieter},
+  journal={Advances in Neural Information Processing Systems},
+  year={2020}
+}
+
+@article{song2021denoising,
+  title={Denoising Diffusion Implicit Models},
+  author={Song, Jiaming and Meng, Chenlin and Ermon, Stefano},
+  journal={International Conference on Learning Representations (ICLR)},
+  year={2021}
+}
+
+@article{song2021score,
+  title={Score-Based Generative Modeling through Stochastic Differential Equations},
+  author={Song, Yang and Sohl-Dickstein, Jascha and Kingma, Diederik P and Kumar, Abhishek and Ermon, Stefano and Poole, Ben},
+  journal={International Conference on Learning Representations (ICLR)},
+  year={2021}
+}
+
+@article{rombach2022high,
+  title={High-Resolution Image Synthesis with Latent Diffusion Models},
+  author={Rombach, Robin and Blattmann, Andreas and Lorenz, Dominik and Esser, Patrick and Ommer, Björn},
+  journal={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
+  year={2022}
+}
+
+@article{ramesh2022hierarchical,
+  title={Hierarchical Text-Conditional Image Generation with CLIP Latents},
+  author={Ramesh, Aditya and Pavlov, Mikhail and Goh, Gabriel and Gray, Scott and Voss, Chelsea and Radford, Alec and Chen, Mark and Sutskever, Ilya},
+  journal={arXiv preprint arXiv:2204.06125},
+  year={2022}
+}
+```
+
+### TorchDiff Repository  
+
+```bibtex
+@misc{torchdiff2025,
+  author = {Samani, Loghman},
+  title = {TorchDiff: A Modular Diffusion Modeling Library in PyTorch},
+  year = {2025},
+  publisher = {GitHub},
+  journal = {GitHub repository},
+  howpublished = {\url{https://github.com/LoqmanSamani/TorchDiff}},
+}
+```
+