Wing Lian d4d84d48af fix ray train and add fsdp2 smoke test for ray trainer (#3053) ...

* add fsdp2 smokle test for ray trainer

* fix raytrain with fsdp2

2025-08-11 09:31:54 -04:00

..

integrations

basic torchao fp8 mixed precision training (#2926)

2025-07-22 16:27:47 -04:00

kernels

Lora kernels bias support (#3025)

2025-08-06 20:20:08 -04:00

multigpu

fix ray train and add fsdp2 smoke test for ray trainer (#3053)

2025-08-11 09:31:54 -04:00

patched

drop old patches and code that are no longer needed (#3007) [skip ci]

2025-08-06 08:02:39 -04:00

solo

jagged lr restart scheudler (#1680) [skip ci]

2025-07-31 13:50:03 -04:00

__init__.py

missing dunder-init

2023-11-06 18:33:01 -05:00

.gitignore

Support Sample packing for phi arch (#586)

2023-09-15 15:46:54 -04:00

test_activation_offloading.py

Act offload lora fix (#2928) [skip ci]

2025-07-24 16:10:04 -04:00

test_deepseekv3.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_dpo.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_embeddings_lr.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_evaluate.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_falcon.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_gemma2.py

Data loader refactor (#2707)

2025-06-10 19:53:07 -04:00

test_gemma3_text.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_imports.py

Fix: RL base feature parity (#2133)

2025-05-30 11:21:47 +07:00

test_llama_pretrain.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_llama_vision.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_llama.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_load_model.py

Distributed/ND-Parallel (#2977)

2025-07-31 15:25:02 -04:00

test_lora_llama.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_mamba.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_mistral.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_mixtral.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_optimizers.py

Dion optimizer support (#3014)

2025-08-04 16:33:30 -04:00

test_packing_loss.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_phi.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_preprocess.py

don't init distributed for deepspeed if preprocessing (#2920)

2025-07-14 14:19:19 -04:00

test_process_reward_model_smollm2.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_profiler.py

allow profiling in mid-training rather from the start (#2899) [skip ci]

2025-07-14 20:11:11 -04:00

test_qat.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_quantization.py

QAT (#2590)

2025-05-28 12:35:47 +01:00

test_qwen.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_reward_model_smollm2.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_save_first_step.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

test_schedulers.py

checkpoint model on first step callback (#2906)

2025-07-15 15:00:48 -04:00

utils.py

Distributed/ND-Parallel (#2977)

2025-07-31 15:25:02 -04:00