What’s Changed

🐜 Bug Fixes

• Update multi-gpu notebook to set cupy device @edknv (#675)
• Fix bug in get_output_sizes_from_schema with core-schema @marcromeyn (#663)
• Remove torch.squeeze() step from the model's forward method. @sararb (#659)
• Set device in dataloaders @edknv (#654)
• Fix the predictions returned by Trainer.predict(..) @sararb (#641)

🚀 Features

• Implemented sampled softmax for NextItemPredictionTask @gabrielspmoreira (#671)
• Add support for ragged inputs to model @oliverholworthy (#666)
• Allow Schema class from core to be used to create Trainer @marcromeyn (#642)
• Cleanup shapes in model.input_schema and output_schema @rnyak (#628)
• Allow Schema class from core to be used to create TabularSequenceFeatures @marcromeyn (#638)
• Allow Schema class from core to be used to create input-blocks @marcromeyn (#634)

📄 Documentation

• extend getting started serving example to serve NVT and TF4Rec model together @rnyak (#670)
• Cropped the table and addressed comments from previous PR @nzarif (#510)
• Update example notebooks to create schema object from merlin core @rnyak (#650)
• replace NVTabulardataloader with Merlindataloader @rnyak (#644)

🔧 Maintenance

• Update multi-gpu notebook to set cupy device @edknv (#675)
• add concurrency setting to stop tests when new commits get pushed to a PR @nv-alaiacano (#673)
• Switch to using 2 GPU action runners for multi-GPU testing @karlhigley (#665)
• Add workflow to check if base branch of pull request is development @oliverholworthy (#656)
• fix the ci script for new unit tests setup @jperez999 (#658)
• Add unit test for serving torchscript model example notebook @rnyak (#657)
• Separate notebook tests into their own tox environment @nv-alaiacano (#653)
• Update usage of use_amp to use_cuda_amp for transformers>=4.20 @oliverholworthy (#627)
• Update example notebooks to create schema object from merlin core @rnyak (#650)
• Update padding of ragged features to enable dataloader change @oliverholworthy (#647)
• fix model output_schema dims for BC/Regression task case @rnyak (#646)
• fix test_remove_consecutive_interactions unit test @rnyak (#643)
• replace NVTabulardataloader with Merlindataloader @rnyak (#644)
• Cleanup shapes in model.input_schema and output_schema @rnyak (#628)
• Migrate schema Tags to merlin.schema.Tags @nv-alaiacano (#632)
• Clean up imports in tests @marcromeyn (#626)