Removed `gpu` parameter (will be added later)