KF Pytorch API reference#

This package contains things that are useful when extending Flytekit.

PyTorch(num_workers)

Configuration for an executable Pytorch Job.