slam_llm/policies/__init__.py

# Copyright (c) Meta Platforms, Inc. and affiliates.
# This software may be used and distributed according to the terms of the Llama 2 Community License Agreement.

from slam_llm.policies.mixed_precision import *
from slam_llm.policies.wrapping import *
from slam_llm.policies.activation_checkpointing_functions import apply_fsdp_checkpointing
from slam_llm.policies.anyprecision_optimizer import AnyPrecisionAdamW