Yichen Zhou
26 lines
524 B
YAML
26 lines
524 B
YAML
name: tfm_env
|
|
|
|
channels:
|
|
- conda-forge
|
|
- defaults
|
|
- anaconda
|
|
dependencies:
|
|
- jupyterlab
|
|
- pip
|
|
- python=3.10
|
|
- pip:
|
|
- datasetsforecast
|
|
- fire
|
|
- git+https://github.com/awslabs/gluon-ts.git
|
|
- huggingface_hub[cli]
|
|
- neuralforecast
|
|
- orjson
|
|
- statsforecast
|
|
- utilsforecast
|
|
- git+https://github.com/amazon-science/chronos-forecasting.git
|
|
- praxis
|
|
- paxml
|
|
- -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
|
|
- "jax[cuda12_pip]"
|
|
- einshape
|