Similar repositories to cgarciae/jax-streams:

• karpathy/nanoGPT github similar
• google-research/tuning_playbook github similar
• huggingface/diffusion-models-class github similar
• danistefanovic/build-your-own-x github similar
• cgarciae/jax-streams github similar
• cgarciae/jax-gravity github similar
• instadeepai/awesome-marl github similar
• galeone/tf-aoc github similar
• lucidrains/flash-attention-jax github similar
• visipedia/herbarium_comp github similar
• Rishit-dagli/Transformer-in-Transformer github similar
• patil-suraj/stable-diffusion-jax github similar
• jax-ml/jax-triton github similar
• cgarciae/treeo github similar
• google/paxml github similar
• henry-prior/jax-rl github similar
• lightonai/RITA github similar
• nd7141/graph_datasets github similar
• ikostrikov/implicit_q_learning github similar
• instadeepai/jumanji github similar
• probml/dynamax github similar
• Joshuaalbert/jaxns github similar
• google/tree-math github similar
• dome272/VQGAN-pytorch github similar
• jax-ml/oryx github similar
• m-bain/webvid github similar
• google-research/longt5 github similar
• araffin/sbx github similar
• google/flaxformer github similar
• rwightman/efficientnet-jax github similar
• lucidrains/PaLM-jax github similar
• radix-ai/awesome-machine-learning-engineer github similar
• vwxyzjn/ppo-implementation-details github similar
• deepmind/dm_pix github similar
• Sea-Snell/JAXSeq github similar
• matthias-wright/flaxmodels github similar
• ikostrikov/walk_in_the_park github similar
• AakashKumarNain/TF_JAX_tutorials github similar
• montrealrobotics/DeepRLInTheWorld github similar
• diambra/arena github similar