Pinned Loading

1. gnn-config-ranker gnn-config-ranker Public

   A graph neural network (GNN) based model which predicts the runtime of deep learning computational graphs on TPUs.

   Python

2. retrieve_and_rank retrieve_and_rank Public

   This repository implements a movie recommendation system using TensorFlow Recommenders (TFRS) with the MovieLens 100K dataset. It features retrieval and ranking models employing a two-tower archite…

   Python 1 1

3. behavioral-sequence-transformer behavioral-sequence-transformer Public

   A recommender system model (RecSys) using a transformer neural network. This model takes a sequence of ratings and predicts what a user will rate a new movie. Trained on the movielens dataset.

   Python 3 1

4. RAG-code-resolution RAG-code-resolution Public

   Proof of concept for automatic fixing of errors in code. A bug report or a stack trace is used as input, then cosine similarity is calculated on the indexed code base (RAG), then an LLM can be used…

   Python 1

5. RL_SB_Agent RL_SB_Agent Public

   This is a deep reinforcement learning agent, using deep Q-learning, to solve the game of Sokoban.

   Python

6. sampleGAN sampleGAN Public

   A text-to-music model. This generative network takes in a phrase as input and generates an audio file.

   Python