Awesome Open Source
Search
Programming Languages
Languages
All Categories
Categories
About
Search results for attention mechanism jax
attention-mechanism
x
jax
x
4 search results found
Palm Jax
⭐
181
Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
Flash Attention Jax
⭐
123
Implementation of Flash Attention in Jax
Travis
⭐
36
TrAVis: Visualise BERT attention in-browser
Fjformer
⭐
10
Embark on a journey of paralleled/unparalleled computational prowess with FJFormer - an arsenal of custom Jax Flax Functions and Utils that elevate your AI endeavors to new heights!
Jax Models
⭐
10
Explore implementations of deep learning concepts like Transformers, Attention, Llama, GPT, InstructGPT, RLHF, Gaussian Processes, Bayesian Inference, Newton Raphson, Distributed Trainers and more!
Related Searches
Python Attention Mechanism (517)
Deep Learning Attention Mechanism (370)
1-4 of 4 search results
Privacy
|
About
|
Terms
|
Follow Us On Twitter
Copyright 2018-2024 Awesome Open Source. All rights reserved.