wbrenton

wbrenton's Stars

• lucidrains/PaLM-jax

  Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)

  Language:Python18412

• google/flax

  Flax is a neural network library for JAX that is designed for flexibility.

  Language:Python5.9k628

• google-deepmind/graphcast

  Language:Python4.5k560

• SakanaAI/AI-Scientist

  The AI Scientist: Towards Fully Automated Open-Ended Scientific Discovery 🧑‍🔬

  Language:Jupyter Notebook7.4k985

• google-deepmind/enn

  Language:Python28955

• yixiaoer/mistral-v0.2-jax

  JAX implementation of the Mistral 7b v0.2 model

  Language:Python322

• Rayhane-mamah/Tacotron-2

  DeepMind's Tacotron-2 Tensorflow implementation

  Language:Python2.3k906

• neonbjb/tortoise-tts

  A multi-voice TTS system trained with an emphasis on quality

  Language:Jupyter Notebook12.9k1.8k

• google-deepmind/rlax

  Language:Python1.2k85

• luchris429/JaxLife

  An Open-Ended Agentic Simulator

  Language:Python161

• jiasenlu/vit-vqgan-jax

  Jax implementation of VIT-VQGAN

  Language:Python71

• google-research/neuralgcm

  Hybrid ML + physics model of the Earth's atmosphere

  Language:Python61265

• mttga/purejaxql

  Simple single-file baselines for Q-Learning in pure-GPU setting

  Language:Python863

• imbue-ai/carbs

  Cost aware hyperparameter tuning algorithm

  Language:Jupyter Notebook936

• trotsky1997/MathBlackBox

  Language:Python46356

• NVIDIA/JAX-Toolbox

  JAX-Toolbox

  Language:Jupyter Notebook23143

• PufferAI/PufferLib

  Simplifying reinforcement learning for complex game environments

  Language:Python1k40

• All-Hands-AI/OpenHands

  🙌 OpenHands: Code Less, Make More

  Language:Python31.3k3.6k

• google/jax

  Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

  Language:Python29.9k2.7k

• Cognitive-AI-Systems/pogema

  POGEMA stands for Partially-Observable Grid Environment for Multiple Agents. This is a grid-based environment that was specifically designed to be flexible, tunable and scalable. It can be tailored to a variety of PO-MAPF settings.

  Language:Python20723

• facebookresearch/LaMCTS

  The release codes of LA-MCTS with its application to Neural Architecture Search.

  Language:Python45770

• chuanyangjin/fast-DiT

  Fast Diffusion Models with Transformers

  Language:Python67090

• kyunghyuncho/map_plan_backprop

  Language:Jupyter Notebook9

• google-research/big_vision

  Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.

  Language:Jupyter Notebook2.2k147

• google-research/vision_transformer

  Language:Jupyter Notebook10.1k1.3k

• borisdayma/clip-jax

  Train vision models using JAX and 🤗 transformers

  Language:Python9110

• vikhyat/moondream

  tiny vision language model

  Language:Jupyter Notebook4.9k435

• sanchit-gandhi/whisper-jax

  JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.

  Language:Jupyter Notebook4.4k368

• huggingface/nanotron

  Minimalistic large language model 3D-parallelism training

  Language:Python1.1k105

• WuTheFWasThat/hanabi.rs

  State of the art Hanabi bots + simulation framework in rust

  Language:Rust4311