From 78d507775f00de9be6895473e1f05e381cc6a335 Mon Sep 17 00:00:00 2001
From: jopetty <research@jacksonpetty.org>
Date: Fri, 17 May 2024 15:04:17 -0400
Subject: [PATCH] Add optional variants for tensor backends

---
 pyproject.toml | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index 2affea0..d460089 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -22,6 +22,18 @@ dependencies = [
 
 [project.optional-dependencies]
 dev = ["pre-commit"]
+torch = ["torch"]
+jax = [
+    "flax",
+    "jax"
+]
+hf = [
+    "accelerate",
+    "datasets",
+    "evaluate",
+    "transformers",
+    "tokenizers",
+]
 
 [tool.ruff]
 line-length = 88