README.md
pyproject.toml
setup.py
ohara/Transformer.py
ohara/__init__.py
ohara/dataset.py
ohara/dpo.py
ohara/inference.py
ohara/lr_scheduler.py
ohara/pretokenize.py
ohara/swa.py
ohara/trainer.py
ohara.egg-info/PKG-INFO
ohara.egg-info/SOURCES.txt
ohara.egg-info/dependency_links.txt
ohara.egg-info/entry_points.txt
ohara.egg-info/requires.txt
ohara.egg-info/top_level.txt
ohara/adaptor/__init__.py
ohara/adaptor/lora.py
ohara/embedings_pos/__init__.py
ohara/embedings_pos/alibi.py
ohara/embedings_pos/rotatry.py
ohara/embedings_pos/xpos.py
ohara/jaxx/__init__.py
ohara/jaxx/attention.py
ohara/jaxx/dev.py
ohara/jaxx/ffn.py
ohara/jaxx/llama.py
ohara/jaxx/norm.py
ohara/jaxx/rope.py
ohara/jaxx/swa.py
ohara/jaxx/train.py
ohara/jaxx/transformer.py
ohara/models/__init__.py
ohara/models/gemma.py
ohara/models/gpt.py
ohara/models/gpt_swiglu.py
ohara/models/llama.py
ohara/models/mamba.py
ohara/models/phi.py
ohara/models/retnet.py
ohara/models/roformer.py
ohara/models/transformer.py
ohara/modules/__init__.py
ohara/modules/activations.py
ohara/modules/attention.py
ohara/modules/base.py
ohara/modules/grouped_query.py
ohara/modules/linear_rnn.py
ohara/modules/mlp.py
ohara/modules/moe.py
ohara/modules/norm.py
ohara/modules/pscan.py
ohara/utils/__init__.py
ohara/utils/generate.py
ohara/utils/info.py
ohara/utils/load.py
ohara/utils/rand.py
ohara/utils/svd.py
ohara/utils/tools.py