logoalt Hacker News

PSSA: A non-transformer language model written from scratch in Rust

59 points • by sparticle62 • today at 3:19 AM • 15 comments • view on HN

Comments

janalsncm • today at 5:01 AM

OP, you should not have written this in Rust. It should be in PyTorch, which is by far the most popular. We can’t tell if this architecture is good or whether there is a problem in your implementation.

You can test the whole thing for free on a GPU with Google Colab. Test both the transformer and your new architecture on a larger dataset. Something that maxes out the GPU for an hour each run.

Also, the readme mentions keeping the same optimizer schedule which sounds nice at first but they are completely different architectures. The loss is high on the transformer, did you try raising the learning rate on it?

In general I’m interested in parameter efficient architectures. I don’t think transformers are optimal, and indeed many improvements have been made to vanilla transformers. But if you have an idea for something better you need to show it.

➕ show 2 replies
saglogog • today at 6:27 AM

Nice idea actually, I always wondered why there was no actual programming in ML!

a19486 • today at 5:14 AM

Homie just had some tokens to burn at the end of the month and “in Rust” is pure HN clickbait.

mllev15 • today at 4:04 AM

You can just tell when the idea itself was generated

Marcuss2 • today at 5:37 AM

How would it compare to current state of the art tested state space layers like Kimi Delta Attention?

kasumispencer2 • today at 5:06 AM

Isn't this just RNN and nearly nothing about this is actually new?

purple-leafy • today at 4:07 AM

“In Rust” man of all the cliche title baits, I hate this one the most.

Why did you choose Rust? Why does that matter?

Nothing wrong with Rust. Lots wrong with the bandwagon that “in Rust” somehow adds value.

➕ show 3 replies
anon291 • today at 4:10 AM

Seems similar to a neural turing machine.

➕ show 1 reply
octoberfranklin • today at 3:59 AM

vibe coded vibe coder!