Luis Vega 0bda1f9780 feat: Nemotron-H model support (#3430) ... * added files for nemotron-h Signed-off-by: Luis Vega <lvega@nvidia.com> * use try/except to import RMSNorm Signed-off-by: Luis Vega <lvega@nvidia.com> --------- Signed-off-by: Luis Vega <lvega@nvidia.com> Co-authored-by: QI JUN <22017000+QiJune@users.noreply.github.com>		2025-04-16 14:05:56 -07:00
..
attention_backend	Support CUDA graphs for EAGLE3 (#3176)	2025-04-17 04:53:50 +08:00
auto_deploy	chore: move all distributed related codes into _torch.distributed directory (#3511)	2025-04-15 08:39:17 +08:00
compilation	feat: Add NVFP4 UB pattern optimization pass in torch compile (#3371)	2025-04-11 21:25:29 +08:00
custom_ops	feat: Support cos_sin_cache in all cases. (#3517)	2025-04-16 13:48:44 +08:00
distributed	chore: move all distributed related codes into _torch.distributed directory (#3511)	2025-04-15 08:39:17 +08:00
models	feat: Nemotron-H model support (#3430)	2025-04-16 14:05:56 -07:00
modules	feat: Nemotron-H model support (#3430)	2025-04-16 14:05:56 -07:00
peft	lora_tests (#3201)	2025-04-09 18:06:52 +03:00
pyexecutor	feat: Nemotron-H model support (#3430)	2025-04-16 14:05:56 -07:00
speculative	Support CUDA graphs for EAGLE3 (#3176)	2025-04-17 04:53:50 +08:00
__init__.py	Update TensorRT-LLM (#2755)	2025-02-11 03:01:00 +00:00
autotuner.py	feat: Apply the new torch-flow compatible AutoTuner to both Fused MoE and NVFP4 Linear operators. (#3151)	2025-04-08 14:28:36 +08:00
llm.py	test: [TRTLLM-4334] Create 1.0 criteria scope from API stability references (#3069)	2025-03-26 18:14:35 +08:00
metadata.py	feat: no-cache attention in PyTorch workflow (#3085)	2025-04-05 01:54:32 +08:00
model_config.py	feat: Support cos_sin_cache in all cases. (#3517)	2025-04-16 13:48:44 +08:00
pipeline_interface.py	Update (#2978)	2025-03-23 16:39:35 +08:00
utils.py	Cache sin cos in model instead of global LRU cache. (#3378)	2025-04-14 11:19:09 +08:00