environment.yml

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
    - jax[cuda12]==0.4.26
    - einshape
    - python-dotenv
    - nixtla>=0.5.1
    - rich