Amazon Web Services
AI Shanghai Lablet

Amazon Web Services
AI Shanghai Lablet

Who Are We?

Amazon Web Services AI Shanghai Lablet was established in the fall of 2018 and is part of Amazon Web Services Machine Learning.
We focus on:
- Open-source projects such as the popular Deep Graph Library (DGL) framework
- Fundamental research in the area of graph neural networks (GNNs) and their applications
- Impact to Amazon Web Services customers through Amazon Web Services releases such as SageMaker DGL and Netpune ML
- Actively foster collaboration with the research community
The Latest News
We have four papers accepted at ACL 2023.
- An AMR-based Link Prediction Approach for Document-level Event Argument Extraction
- Exploiting Abstract Meaning Representation for Open-Domain Question Answering
- Dual Cache for Long Document Neural Coreference Resolution
- Distributed Marker Representation for Ambiguous Discourse Markers and Entangled Relations

We are thrilled to announce the arrival of DGL 1.0, a cutting-edge machine learning framework for deep learning on graphs.

In the latest DGL v0.9.1, we released a new pipeline for preprocess, partition and dispatch graph of billions of nodes or edges for distributed GNN training. At its core is a new data format called Chunked Graph Data Format (CGDF) which stores graph data by chunks. The new pipeline processes data chunks in parallel which not only reduces the memory requirement of each machine but also significantly accelerates the entire procedure.

In the latest issue of Nature Machine Intelligence, Yousun Jung and his research team at Korea Institute of Science and Technology (KAIST) have proposed a novel generalized-template-based graph neural network called LocalTransform for accurate organic reactivity prediction . The generalized reaction templates demonstrate the built-in interpretability in addition to state-of-the-art prediction accuracy. LocalTransform has been implemented with DGL, DGL-LifeSci and PyTorch.

Six years after the first Graph Convolutional Networks paper, researchers are actively investigating more advanced GNN architecture or training methodology. As the developer team of DGL, we closely watch those new research trends and release features to facilitate them. Here, we're excited to announce a new major release DGL v0.9.

The new 0.8.2 update brings new additions for the GNN community.

V0.8.1 is a minor release that includes module updates, optimizations, new features and bug fixes.

We are excited to announce the release of DGL v0.8, which brings many new features as well as improvement on system performance.

Today, the GAMMA Lab of Beijing University of Posts and Telecommunications officially released OpenHGNN-0.1.0, an Open Source Toolkit for Heterogeneous Graph Neural Network. OpenHGNN is based on DGL [Deep Graph Library] and PyTorch, and includes SOTA models of heterogeneous graph.

We Provide Open Source Projects for Researchers and Developers

DGL
DGL is an easy-to-use, high performance and scalable Python package for deep learning on graphs. DGL is framework agnostic, meaning if a deep graph model is a component of an end-to-end application, the rest of the logics can be implemented in any major frameworks, such as PyTorch, Apache MXNet or TensorFlow.

DGL-LifeSci
DGL-LifeSci is a specialized package for applications in bioinformatics and cheminformatics powered by graph neural networks.

DGL-KE
DGL-KE is an easy-to-use and highly scalable package for learning large-scale knowledge graph embeddings.
We Provide Open Source Projects for Researchers and Developers

DGL
DGL is an easy-to-use, high performance and scalable Python package for deep learning on graphs. DGL is framework agnostic, meaning if a deep graph model is a component of an end-to-end application, the rest of the logics can be implemented in any major frameworks, such as PyTorch, Apache MXNet or TensorFlow.

DGL-LifeSci
DGL-LifeSci is a specialized package for applications in bioinformatics and cheminformatics powered by graph neural networks.

DGL-KE
DGL-KE is an easy-to-use and highly scalable package for learning large-scale knowledge graph embeddings.
Publications

See publications from Amazon Web Services AI Shanghai Lablet.
Career Opportunities
Amazon Web Services AI Shanghai Lablet is hiring year-round intern positions. Please visit Amazon Jobs to learn more about available positions and job descriptions. Interested candidates are welcome to apply via the position link.
