pytorch-lightning
ray
