You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/docs/colossalai
Ziyue Jiang 63f250bbd4
fix file name (#1759)
2 years ago
..
colossalai.amp.amp_type.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.amp.apex_amp.apex_amp.rst update rst (#615) 3 years ago
colossalai.amp.apex_amp.rst update rst (#615) 3 years ago
colossalai.amp.naive_amp.grad_scaler.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.amp.naive_amp.naive_amp.rst update rst (#615) 3 years ago
colossalai.amp.naive_amp.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.amp.rst update rst (#615) 3 years ago
colossalai.amp.torch_amp.rst update rst (#615) 3 years ago
colossalai.amp.torch_amp.torch_amp.rst update rst (#615) 3 years ago
colossalai.builder.builder.rst Migrated project 3 years ago
colossalai.builder.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.cli.benchmark.benchmark.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.cli.benchmark.models.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.cli.benchmark.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.cli.benchmark.utils.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.cli.check.check_installation.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.cli.check.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.cli.cli.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.cli.launcher.hostinfo.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.cli.launcher.multinode_runner.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.cli.launcher.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.cli.launcher.run.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.cli.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.communication.collective.rst Migrated project 3 years ago
colossalai.communication.p2p.rst Migrated project 3 years ago
colossalai.communication.ring.rst Migrated project 3 years ago
colossalai.communication.rst update rst (#615) 3 years ago
colossalai.communication.utils.rst Migrated project 3 years ago
colossalai.constants.rst update rst (#615) 3 years ago
colossalai.context.config.rst Migrated project 3 years ago
colossalai.context.moe_context.rst update rst (#615) 3 years ago
colossalai.context.parallel_context.rst Migrated project 3 years ago
colossalai.context.parallel_mode.rst Migrated project 3 years ago
colossalai.context.process_group_initializer.initializer_1d.rst Migrated project 3 years ago
colossalai.context.process_group_initializer.initializer_2d.rst update rst (#615) 3 years ago
colossalai.context.process_group_initializer.initializer_2p5d.rst update rst (#615) 3 years ago
colossalai.context.process_group_initializer.initializer_3d.rst update rst (#615) 3 years ago
colossalai.context.process_group_initializer.initializer_data.rst Migrated project 3 years ago
colossalai.context.process_group_initializer.initializer_model.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.context.process_group_initializer.initializer_pipeline.rst Migrated project 3 years ago
colossalai.context.process_group_initializer.initializer_sequence.rst update rst (#615) 3 years ago
colossalai.context.process_group_initializer.initializer_tensor.rst Migrated project 3 years ago
colossalai.context.process_group_initializer.process_group_initializer.rst Migrated project 3 years ago
colossalai.context.process_group_initializer.rst update rst (#615) 3 years ago
colossalai.context.random.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.context.random.seed_manager.rst Migrated project 3 years ago
colossalai.context.rst update rst (#615) 3 years ago
colossalai.context.singleton_meta.rst update rst (#615) 3 years ago
colossalai.core.rst update rst (#615) 3 years ago
colossalai.engine.gradient_accumulation.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.engine.gradient_handler.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.engine.gradient_handler.utils.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.engine.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.engine.schedule.rst Migrated project 3 years ago
colossalai.fx.passes.adding_split_node_pass.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.fx.passes.meta_info_prop.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.fx.passes.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.fx.passes.shard_1d_pass.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.fx.passes.split_module.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.fx.passes.utils.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.fx.proxy.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.fx.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.fx.tracer.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.fx.tracer.tracer.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.chunk.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.chunk_mgr.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.gemini_context.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.gemini_mgr.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.memory_tracer.memory_monitor.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.memory_tracer.memstats_collector.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.memory_tracer.model_data_memtracer.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.memory_tracer.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.ophooks.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.ophooks.utils.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.paramhooks.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.placement_policy.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.stateful_tensor.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.stateful_tensor_container.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.stateful_tensor_mgr.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.tensor_placement_policy.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.gemini.tensor_utils.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.global_variables.rst update rst (#615) 3 years ago
colossalai.initialize.rst Migrated project 3 years ago
colossalai.kernel.cuda_native.layer_norm.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.kernel.cuda_native.multihead_attention.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.kernel.cuda_native.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.kernel.cuda_native.scaled_softmax.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.kernel.jit.bias_dropout_add.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.kernel.jit.bias_gelu.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.kernel.jit.option.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.kernel.jit.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.kernel.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.logging.logger.rst update rst (#615) 3 years ago
colossalai.logging.rst update rst (#615) 3 years ago
colossalai.nn.graph.graph_node.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.nn.graph.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.nn.graph.utils.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.nn.init.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.nn.layer.base_layer.rst Migrated project 3 years ago
colossalai.nn.layer.colossalai_layer.dropout.rst update rst (#615) 3 years ago
colossalai.nn.layer.colossalai_layer.embedding.rst update rst (#615) 3 years ago
colossalai.nn.layer.colossalai_layer.linear.rst update rst (#615) 3 years ago
colossalai.nn.layer.colossalai_layer.normalization.rst update rst (#615) 3 years ago
colossalai.nn.layer.colossalai_layer.rst update rst (#615) 3 years ago
colossalai.nn.layer.moe.experts.rst update rst (#615) 3 years ago
colossalai.nn.layer.moe.layers.rst update rst (#615) 3 years ago
colossalai.nn.layer.moe.rst update rst (#615) 3 years ago
colossalai.nn.layer.moe.utils.rst update rst (#615) 3 years ago
colossalai.nn.layer.parallel_1d.layers.rst update rst (#615) 3 years ago
colossalai.nn.layer.parallel_1d.rst update rst (#615) 3 years ago
colossalai.nn.layer.parallel_2d.layers.rst update rst (#615) 3 years ago
colossalai.nn.layer.parallel_2d.rst update rst (#615) 3 years ago
colossalai.nn.layer.parallel_2p5d.layers.rst update rst (#615) 3 years ago
colossalai.nn.layer.parallel_2p5d.rst update rst (#615) 3 years ago
colossalai.nn.layer.parallel_3d.layers.rst update rst (#615) 3 years ago
colossalai.nn.layer.parallel_3d.rst update rst (#615) 3 years ago
colossalai.nn.layer.parallel_sequence.layers.rst update rst (#615) 3 years ago
colossalai.nn.layer.parallel_sequence.rst update rst (#615) 3 years ago
colossalai.nn.layer.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.nn.layer.utils.common.rst update rst (#615) 3 years ago
colossalai.nn.layer.utils.rst update rst (#615) 3 years ago
colossalai.nn.layer.vanilla.layers.rst update rst (#615) 3 years ago
colossalai.nn.layer.vanilla.rst update rst (#615) 3 years ago
colossalai.nn.layer.wrapper.pipeline_wrapper.rst update rst (#615) 3 years ago
colossalai.nn.layer.wrapper.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.nn.loss.loss_1d.rst [doc] update rst (#470) 3 years ago
colossalai.nn.loss.loss_2d.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.nn.loss.loss_2p5d.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.nn.loss.loss_3d.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.nn.loss.loss_moe.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.nn.loss.rst [doc] update rst (#470) 3 years ago
colossalai.nn.lr_scheduler.cosine.rst Migrated project 3 years ago
colossalai.nn.lr_scheduler.delayed.rst Migrated project 3 years ago
colossalai.nn.lr_scheduler.linear.rst Migrated project 3 years ago
colossalai.nn.lr_scheduler.multistep.rst Migrated project 3 years ago
colossalai.nn.lr_scheduler.onecycle.rst Migrated project 3 years ago
colossalai.nn.lr_scheduler.poly.rst Migrated project 3 years ago
colossalai.nn.lr_scheduler.rst update rst (#615) 3 years ago
colossalai.nn.lr_scheduler.torch.rst Migrated project 3 years ago
colossalai.nn.metric.accuracy_2d.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.nn.metric.accuracy_2p5d.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.nn.metric.accuracy_3d.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.nn.metric.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.nn.optimizer.colossalai_optimizer.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.nn.optimizer.cpu_adam.rst [doc] update rst (#470) 3 years ago
colossalai.nn.optimizer.fused_adam.rst Migrated project 3 years ago
colossalai.nn.optimizer.fused_lamb.rst Migrated project 3 years ago
colossalai.nn.optimizer.fused_sgd.rst Migrated project 3 years ago
colossalai.nn.optimizer.hybrid_adam.rst [docs] updatad docs of hybrid adam and cpu adam (#552) 3 years ago
colossalai.nn.optimizer.lamb.rst Migrated project 3 years ago
colossalai.nn.optimizer.lars.rst update documentation 3 years ago
colossalai.nn.optimizer.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.nn.optimizer.utils.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.nn.parallel.data_parallel.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.nn.parallel.layers.colo_module.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.nn.parallel.layers.embedding.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.nn.parallel.layers.linear.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.nn.parallel.layers.module_utils.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.nn.parallel.layers.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.nn.parallel.reducer.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.nn.parallel.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.nn.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.pipeline.layer_sepc.rst fix file name (#1759) 2 years ago
colossalai.pipeline.pipelinable.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.pipeline.rst fix file name (#1759) 2 years ago
colossalai.pipeline.utils.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.registry.registry.rst Migrated project 3 years ago
colossalai.registry.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.tensor.colo_parameter.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.tensor.colo_tensor.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.tensor.compute_spec.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.tensor.const.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.tensor.dist_spec_mgr.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.tensor.distspec.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.tensor.op_wrapper.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.tensor.param_op_hook.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.tensor.process_group.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.tensor.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.tensor.tensor_spec.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.tensor.utils.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.testing.comparison.rst update rst (#615) 3 years ago
colossalai.testing.rst update rst (#615) 3 years ago
colossalai.testing.utils.rst update rst (#615) 3 years ago
colossalai.trainer.hooks.rst Migrated project 3 years ago
colossalai.trainer.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.utils.activation_checkpoint.rst Migrated project 3 years ago
colossalai.utils.checkpoint.module_checkpoint.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.checkpoint.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.checkpoint.utils.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.checkpointing.rst Support TP-compatible Torch AMP and Update trainer API (#27) 3 years ago
colossalai.utils.common.rst Migrated project 3 years ago
colossalai.utils.cuda.rst Migrated project 3 years ago
colossalai.utils.data_sampler.base_sampler.rst update rst (#615) 3 years ago
colossalai.utils.data_sampler.data_parallel_sampler.rst update rst (#615) 3 years ago
colossalai.utils.data_sampler.rst update rst (#615) 3 years ago
colossalai.utils.memory.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.model.colo_init_context.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.model.lazy_init_context.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.model.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.model.utils.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.moe.rst update rst (#615) 3 years ago
colossalai.utils.multi_tensor_apply.multi_tensor_apply.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.utils.multi_tensor_apply.rst removed tutorial markdown and refreshed rst files for consistency 3 years ago
colossalai.utils.profiler.extention.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.profiler.legacy.comm_profiler.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.profiler.legacy.mem_profiler.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.profiler.legacy.pcie_profiler.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.profiler.legacy.prof_utils.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.profiler.legacy.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.profiler.profiler.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.profiler.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.profiler.stateful_tensor_mem_extention.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.utils.tensor_detector.rst update rst (#615) 3 years ago
colossalai.utils.tensor_detector.tensor_detector.rst update rst (#615) 3 years ago
colossalai.utils.timer.rst Migrated project 3 years ago
colossalai.zero.init_ctx.init_context.rst update rst (#615) 3 years ago
colossalai.zero.init_ctx.rst update rst (#615) 3 years ago
colossalai.zero.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.zero.shard_utils.base_shard_strategy.rst update rst (#615) 3 years ago
colossalai.zero.shard_utils.bucket_tensor_shard_strategy.rst update rst (#615) 3 years ago
colossalai.zero.shard_utils.commons.rst update rst (#615) 3 years ago
colossalai.zero.shard_utils.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.zero.shard_utils.tensor_shard_strategy.rst update rst (#615) 3 years ago
colossalai.zero.sharded_model.reduce_scatter.rst update rst (#615) 3 years ago
colossalai.zero.sharded_model.rst update rst (#615) 3 years ago
colossalai.zero.sharded_model.sharded_model_v2.rst update rst (#615) 3 years ago
colossalai.zero.sharded_model.utils.rst update rst (#615) 3 years ago
colossalai.zero.sharded_optim.rst update rst (#615) 3 years ago
colossalai.zero.sharded_optim.sharded_optim_v2.rst update rst (#615) 3 years ago
colossalai.zero.sharded_param.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.zero.sharded_param.sharded_param.rst update rst (#615) 3 years ago
colossalai.zero.sharded_param.sharded_tensor.rst update rst (#615) 3 years ago
colossalai.zero.utils.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.zero.utils.zero_hook.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.zero.utils.zero_hook_v2.rst [doc] update rst and docstring (#1351) 2 years ago
colossalai.zero.zero_optimizer.rst [doc] update rst and docstring (#1351) 2 years ago