AlexNet

Flax NNX is a neural-network API for JAX developed by the Flax team in collaboration with the JAX team. Released in 2024, it uses Python reference semantics so models can be expressed as regular Python objects with reference sharing and mutability, supporting inspection, debugging, and analysis. Its API includes layers and modules such as Linear, convolution, normalization, attention, recurrent cells, and dropout, along with patterns for replicated training, serialization, and checkpointing. In distributed training, it can manage model state while JAX handles computation across a device mesh.

Google DeepMind is an artificial-intelligence research and development organization that builds AI models and systems for general use, science, robotics, speech, image and video generation, music, and coding. Its featured systems include the Gemini family, Gemma open models, Veo video generation, Lyria music generation, Gemini Robotics, and AI systems for scientific discovery. The site also publishes research and provides access to models through products and developer platforms such as Google AI Studio, the Gemini API, and related enterprise tools.

NextTransformer is an open-source research framework, benchmark, and auto-research platform for comparing next-generation Transformer and language-model architectures. It trains recurrent-depth, sparse-attention, linear-attention, residual-stream, and hybrid methods on the same data under a fixed compute-matched protocol, then evaluates them at shared checkpoints against a dense causal-attention baseline. The framework records results as audited, file-backed runs and provides an in-browser observatory with learning curves, replayable leaderboards, architecture diagrams, model cards, and CORE task evaluations. Researchers can add methods through a plugin, calibration, formal training run, audit, and pull-request workflow; the repository also documents an experimental arena for agent-assisted architecture research. It is implemented for PyTorch training on Linux systems with NVIDIA GPUs and is released under the MIT License.

Vertex AI is Google Cloud’s managed platform for developing and deploying machine-learning and generative-AI applications. It provides hosted models and APIs that applications, including agents, can call at runtime, along with tools for model training, evaluation, and serving.

An AI framework for defining model architectures and supporting model serving.