diff --git a/setup.py b/setup.py
index a996431..3c29ba3 100644
--- a/setup.py
+++ b/setup.py
@@ -17,14 +17,14 @@ def func(m):
 
 setuptools.setup(
     name="neurodiffeq",
-    version="0.4.0",
+    version="0.5.0",
     author="neurodiffgym",
     author_email="shuheng_liu@g.harvard.edu",
     description="A light-weight & flexible library for solving differential equations using neural networks based on PyTorch. ",
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/NeuroDiffGym/neurodiffeq",
-    download_url="https://github.com/NeuroDiffGym/neurodiffeq/archive/v0.4.0.tar.gz",
+    download_url="https://github.com/NeuroDiffGym/neurodiffeq/archive/v0.5.0.tar.gz",
     keywords=[
         "neural network",
         "deep learning",