Making models go brrr | Engineering @reflection_ai | Occasional PufferLib contributor

Joined March 2024
Excited to finally share our progress in developing a reinforcement learning system to beat Pokémon Red. Our system successfully completes the game using a policy under 10M parameters, PPO, and a few novel techniques. Blog posted below
12
31
17
401
56,439
Final Fantasy XII's battle system was 10 years ahead of the zeitgeist. It should be remade, but backed by AI agents.
2
130
Open weights are free weights. Open weights are free weights. Therefore, with Beam, everyday can be gym day!
Beam is now available across platforms, with open weights under the Apache 2.0 license - in collaboration with an ecosystem of 40+ partners. We're releasing: - Weights, tokenizer, and reference implementation - BF16, FP8 and NVFP4 weights and a DFlash2 drafter - The open source harness we built and trained Beam on: Mirror CLI Weights and model card on Hugging Face: huggingface.co/reflection
4
28
588
I'll admit the first post was not token efficient. But Beam is!
14
At @reflection_ai, I've used Beam to make a search and recommendation engine for research infrastructure and even a harness to play video games!
14
A quick Reflection: I remember sitting down and monitoring day one of the pretraining run. It's AWESOME to see what Beam became from that day.
Introducing Beam: a highly efficient agentic open model with 501B total parameters and 23B active. - Frontier reasoning efficiency - Advances the Western open frontier on coding & agentic tasks - Trained end-to-end from scratch Full weights release this month. Learn more about Beam: reflection.ai/beam
2
40
814
I like to tell people that when working in research infra, you should expect all your assumptions to last weeks at best
Presenting my grand unified theory of ML researcher impact: Your impact is directly proportional to how much pain you cause to infra. Fundamentally, you can only inflict pain upon infra if your approach actually works. And the better your approach works the more pain infra is forced to endure. So, to give some examples: - MoE's add a ton of data-dependent computation => pain (shazeer++) - GDN/KDA are the most complex architecture I've been forced to care about and a very annoying matrix inversion => pain (sonta++) - Muon is much more annoying than Adam and causes annoying restrictions on parallelism => pain (keller/jeremy++) - RL scaling forced many researchers to care about LLM inference and RL infra as a category => pain (tworek++) Even papers like Attention Is All You Need have lead to significant pain! Before transformers were invented everyone was running small jobs and I never needed to think about kv-caches or 6D parallelism.
1
9
671
My current use case is to profile speedruns for a tool I'm working on. Profiling has helped find inefficient navigation subroutines. It's great.
2
80
A few years ago, I wanted to find a way to maximize shade during my long runs in the summer. Recently, I decided to experiment with a solution. Presenting Shady Route Finder.
3
1
1
5
1,841
shadyroutefinder.com . Supports 8 cities, can tell you what side of the street to walk on, mobile and even has a sunny route finder mode on desktop. I've tested some routes, but obviously not all. Any feedback would be great.
1
1
198
Move over nvidia-smi, ibstat is my new best friend
2
2
273
I dared him to try an ai assisted native rewrite. As expected, 10k sps at best has now become 4M sps. Nice.
Replying to @DanAdvantage
i did start with a rudimentary implementation of pokemon stemming from a native rewrite of pokemon firered. the starting point i used gets around 4,000,000 steps per second as an rl env. here is the entire prompt (caution: long!!!):
2
2
324
대박! A year ago we announced our series A. Today we’re announcing an amazing partnership with Shinsegae. Who knows what’ll come next?
Reflection is partnering with Shinsegae Group to build a 250-megawatt sovereign AI factory for the Republic of Korea. Open intelligence. Built on trust between allies. Owned by the nations that need it most. The future of sovereign AI. Read more in the @WSJ.
1
14
399
Underrated: Letting a coding agent run when you're in meetings.
2
171
Had some fun helping out @kywch500 and @jsuarez simplifying Pufferlib's 2048 env the last couple of weeks. 2x better results with fewer observations, rewards and a new model architecture!
2
3
1
16
6,655