its available to work on a GPU?

[Pandelo0398]

Im regarding some information and i want to know if it is possible to work this library on a GPU

[jmoralez]

Hey. The training and inference can both be done on GPU. It should automatically do so if you have a CUDA GPU available, but you can also provide the accelerator argument to the model constructor with something like "gpu" or "cuda:1".