google/trax
View on GitHub ↗ ★ 8,301 Python archived
Trax — Deep Learning with Clear Code and Speed
8 mentions on Hacker News by 7 people: 2 stories, 6 comments · first 2019-12-08 · latest 2024-04-24
20082026
-
…were some of the best developed communities in the Jax AI field Perhaps the “trax” repo? https://github.com/google/trax Some HF examples https://github.com/huggingface/transformers/tree/main/exampl... Sadly it seems much of the work is…
comment by bionhoward · 2024-04-24 · in Maxtext: A simple, performant and scalable Jax LLM -
…product. Here is the paper https://arxiv.org/abs/2111.12763 and the implementation https://github.com/google/trax/blob/master/trax/models/resea... if you are interested. Hope you get to look into this!
comment by kir-gadjello · 2023-05-03 · in Replit's new Code LLM: Open Source, 77% smaller than Codex, trained in 1 week -
…(specifically, "Terraformer" from this paper https://arxiv.org/abs/2111.12763 and this codebase https://github.com/google/trax/blob/master/trax/models/resea... by Google Brain and OpenAI)? I understand the pressure to focus on training a…
comment by kir-gadjello · 2023-04-18 · in RedPajama: Reproduction of LLaMA with friendly license -
…https://github.com/openai/blocksparse/blob/master/examples/t... https://github.com/google/trax/blob/master/trax/models/trans...
comment by ypcx · 2020-08-17 · in Karpathy's MinGPT -
What about Trax ( https://github.com/google/trax ) and how does it compare with Flax or Haiku? Interesting that googlers who are supposed to use Tensorflow are now actively developing a new autograd engine and at least three new DL…
comment by p1esk · 2020-04-08 · in Jax – Composable transformations of Python and NumPy programs -
…network libraries in JAX. (e.g., https://github.com/google/jax/blob/master/jax/experimental/s... , https://github.com/google/trax ).
comment by nihilest · 2019-12-08 · in Tensorflow User Experience