[README]

README.md

@@ -9,7 +9,7 @@ Implementation of Switch Transformers from the paper: "Switch Transformers: Scal
 ## Installation
 
 ```bash
-pip install -e .
+pip install switch-transformers
 ```
 
 # Usage

pyproject.toml

@@ -3,15 +3,15 @@ requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
-name = "paper"
+name = "switch-transformers"
 version = "0.0.1"
-description = "Paper - Pytorch"
+description = "SwitchTransformers - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <[email protected]>"]
-homepage = "https://github.com/kyegomez/paper"
-documentation = "https://github.com/kyegomez/paper"  # Add this if you have documentation.
+homepage = "https://github.com/kyegomez/SwitchTransformers"
+documentation = "https://github.com/kyegomez/SwitchTransformers"  # Add this if you have documentation.
 readme = "README.md"  # Assuming you have a README.md
-repository = "https://github.com/kyegomez/paper"
+repository = "https://github.com/kyegomez/SwitchTransformers"
 keywords = ["artificial intelligence", "deep learning", "optimizers", "Prompt Engineering"]
 classifiers = [
     "Development Status :: 4 - Beta",
@@ -25,9 +25,9 @@ classifiers = [
 python = "^3.6"
 swarms = "*"
 zetascale = "*"
-
-[tool.poetry.dev-dependencies]
-# Add development dependencies here
+einops = "*"
+torch = "*"
+torchvision = "*"
 
 
 [tool.poetry.group.lint.dependencies]