megatron.memory.get_mem_buff#

megatron.memory.get_mem_buff(name)#

Get the memory buffer.