<- Back
Comments (25)
- k__I was yesterday years old when I learned that those open weight models need custom code to run.Somehow I expected inference engines are generic LLM runtimes that can execute any weight.So, to get this right.Someone trains a model.They release the weights and a reference implementation of the model architecture.Then a provider has to host this model either by running inference via the reference implementation, an open source implementation, or build their own.Does this mean, providers don't just differ in quantisation and configuration, but also in inference engine implementation?
- anuj0456I have been studying modern LLM architectures and started implementing them from scratch in PyTorch to better understand the design choices behind each model.OpenArch is a collection of these implementations, including Llama, Qwen, DeepSeek, Gemma, Kimi, GPT-OSS and others.The goal is to keep the code readable and useful as a reference when going from the paper to an actual implementation.Would be interested in feedback from people working on model architecture and training.
- swimwiththebeatThis is really cool, great way to reinforce our understanding of model architectures! But how is the author confirming that these model architecture implementations are correct though? I don't see any details in the README.md.
- tweedler290This is really cool!Had a question on the MoE: in kimmi-K2/model.py, the router does torch.topk(..., k=self.num_experts) while every other MoE uses k=self.top_k. The ctor's top_k=8 is never stored either.. Is that intentional dense routing or should that be self.top_k?
- theGeatZhopaHey anujThis is excellent for understanding. I'm having some trouble to get into understanding - pytorch is for me the RL which is used as gym/training. There I can chose ppo, dnq and other agents to perform some predefined actions in a predefined gym/world.The repo you are showing - I really have problems to get it into RL understanding of mine. What's the gym? What are the agents. Can it be used to train that models with pytorch?Sorry for the noob question. Papers are overwhelming my noob brain.
- appyjain21One thing i am hoping can make easier understanding how architectural changes trasnlate into actual implementation and performance
- sarra01hey, Building these from scratch in pure PyTorch is honestly the best way to deeply understand the paper details. something better than simply implementing a traditional Transformer or GPT-2，As an individual maintainer, will be able to keep up with future model updates?
- zhubaohua168[flagged]