move requirements/ to studio/backend/ and update paths in setup.sh

studio/backend/requirements/base.txt

@@ -0,0 +1,3 @@
+# Core unsloth packages
+unsloth-zoo
+unsloth

studio/backend/requirements/extras-no-deps.txt

@@ -0,0 +1,13 @@
+# Audio extras (installed with --no-deps --no-cache-dir)
+descript-audio-codec
+descript-audiotools
+julius
+torchcodec
+snac
+
+# TRL and related packages
+trl==0.23.1
+git+https://github.com/meta-pytorch/OpenEnv.git
+executorch==1.0.1
+torch-c-dlpack-ext
+sentence_transformers==5.2.0

studio/backend/requirements/extras.txt

@@ -0,0 +1,56 @@
+# OpenEnv dependencies
+tomli
+tomli-w
+
+# ExecuTorch dependencies
+ruamel.yaml
+coremltools
+expecttest
+flatbuffers
+hydra-core
+hypothesis
+kgb
+parameterized
+pytest<9.0
+pytest-json-report
+pytest-rerunfailures==15.1
+pytest-xdist
+# Also needed by sentence_transformers
+scikit-learn==1.7.1
+
+# Additional extras
+pybind11
+langid
+jiwer
+omegaconf
+einx
+pyloudnorm
+openai-whisper
+uroman
+MeCab
+loguru
+flatten_dict
+ffmpy
+randomname
+argbind
+tiktoken
+ftfy
+importlib-resources
+librosa
+markdown2
+matplotlib
+pystoi
+soundfile
+tensorboard
+torch-stoi
+evaluate
+timm
+transformers-cfg
+open_spiel
+addict
+easydict
+einops
+tabulate
+fastmcp>=2.0.0
+openai>=2.7.2
+websockets>=13.0,<14

studio/backend/requirements/overrides.txt

@@ -0,0 +1,7 @@
+# Torch AO overrides (installed with --force-reinstall --no-cache-dir)
+torchao==0.14.0
+transformers==4.57.1
+pytorch_tokenizers
+
+# Kernel packages
+kernels

studio/backend/requirements/studio.txt

@@ -0,0 +1,13 @@
+# Studio UI backend dependencies
+typer
+fastapi
+uvicorn
+pydantic
+matplotlib
+pandas
+nest_asyncio
+datasets==4.3.0
+pyjwt
+easydict
+addict
+gradio>=4.0.0

studio/backend/requirements/triton-kernels.txt

@@ -0,0 +1,2 @@
+# Triton kernels (installed with --no-deps, from source)
+triton_kernels @ git+https://github.com/triton-lang/triton.git@release/3.6.x#subdirectory=python/triton_kernels