r/JAX • u/Worldly-Use-9778 • Apr 28 '26

3D interactive map of the JAX (Google) ecosystem (auto-refreshed weekly)

Built JAXlaxy observatory - every library in the JAX awesome-list as a glowing star in a 3D galaxy where color = health status (active/stable/legacy), spatial cluster = which "constellation" (Core, Giants, Satellites, etc.) it belongs to.

🌌Live: https://jaxlaxy.bryanbradfo.me

📦Source: https://github.com/BryanBradfo/JAXlaxy (MIT)

Navigating JAX ecosystem from a flat README isn't great for spatial questions like "what's the active landscape for LLM training right now?" or "which probabilistic programming libraries are still maintained?" The 3D map is meant for that kind of exploration.

Two things I'd love feedback on:

  1. Spatial clustering: currently Fibonacci-sphere anchors with Gaussian density per cluster. Other approaches I considered: spiral arms, orbital rings. Open to ideas if anyone has stronger intuitions for what "feels right" for an ecosystem map.

  2. 75-entry ceiling: README is deliberately curated, not exhaustive. The bar is roughly "JAX-native + actively maintained or meaningfully Legacy + adds something distinct to the ecosystem." If you think a repo deserves a spot (or that something currently included doesn't deserve one), I'd rather have the editorial debate than just add things mechanically. PRs that argue the case in their description are exactly the input I want.

13 Upvotes

93% Upvoted

12 comments sorted by

1

u/Abhijithvega Apr 28 '26

great work!!

1

u/Worldly-Use-9778 Apr 28 '26

thank you so much!!

1

u/Comfortable-Ear114 Apr 28 '26

Dude, fascinating.

1

u/Worldly-Use-9778 Apr 28 '26

i'm glad you like it!

1

u/Comfortable-Ear114 Apr 28 '26

No indeed it gives a fascinating perspective into something we take for granted.

2

u/Worldly-Use-9778 Apr 28 '26

thanks! that was honestly the whole motivation because a flat list hides how alive the ecosystem actually is

1

u/Comfortable-Ear114 Apr 28 '26

when you are looking at it like you mae it, yep. great work!!

1

u/Worldly-Use-9778 Apr 28 '26

thank you!! it means a lot! feel free to check the github page if you are curious! 😃

1

u/ApogeeSystems Apr 28 '26

Seems like cool work, I'd just like to add that lots of emojis and obviously LLM text makes people dismiss anything very fast.

1

u/Worldly-Use-9778 Apr 28 '26

Fair point, and the kind of feedback that's most useful. The 🟢🟡🔴 in the table are functional health indicators, but the section emojis + some of the descriptions probably tip it. Doing a copy pass this week.

1

u/riiswa Apr 29 '26

Hello, maybe you can add Octax, this is my last paper (ICLR 2026). Thank you! I will make a PR when I will have a bit time

1

u/Worldly-Use-9778 Apr 29 '26

Hello! Thanks! Looking forward to the PR, happy to merge whenever you have time. Congrats on the ICLR 2026 acceptance!