Sea-Snell's Stars
brendenlake/SCAN
Simple language-driven navigation tasks for studying compositional learning
brentyi/tyro
Zero-effort CLI interfaces & config objects, from types
salesforce/CodeGen
CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.
jlin816/dotfiles
pallets/flask
The Python micro framework for building web applications.
uqfoundation/dill
serialize all of Python
google/ml_collections
ML Collections is a library of Python Collections designed for ML use cases.
CompVis/stable-diffusion
A latent text-to-image diffusion model
alpa-projects/alpa
Training and serving large-scale neural networks with auto parallelization.
cvxgrp/cvxpylayers
Differentiable convex optimization layers
microsoft/VQ-Diffusion
Official implementation of VQ-Diffusion
allenai/natural-instructions
Expanding natural instructions
yizhongw/Tk-Instruct
Tk-Instruct is a Transformer model that is tuned to solve many NLP tasks by following instructions.
allenai/ScienceWorld
ScienceWorld is a text-based virtual environment centered around accomplishing tasks from the standardized elementary science curriculum.
borisdayma/dalle-mini
DALL·E Mini - Generate images from a text prompt
google/gin-config
Gin provides a lightweight configuration framework for Python
google-research/t5x
mpi4jax/mpi4jax
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
patrick-kidger/jaxtyping
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
Dao-AILab/flash-attention
Fast and memory-efficient exact attention
state-spaces/s4
Structured state space sequence models
RobertTLange/gymnax
RL Environments in JAX 🌍
huggingface/diffusers
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
google-deepmind/chex
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
joshuacnf/paradox-learning2reason
conda-forge/miniforge
A conda-forge distribution.
apple/ml-ane-transformers
Reference implementation of the Transformer architecture optimized for Apple Neural Engine (ANE)
CompVis/latent-diffusion
High-Resolution Image Synthesis with Latent Diffusion Models
batra-mlp-lab/visdial-rl
PyTorch code for Learning Cooperative Visual Dialog Agents using Deep Reinforcement Learning