numpy
torch
gymnasium