Efficient Triton Kernels for LLM Training

Repository for "Uncertainty-Aware Machine Translation Evaluation", accepted to Findings of EMNLP 2021.

An Apache 2.0 fork of HuggingFace's Large Language Model Text Generation Inference

🎓 Path to a free self-taught education in Computer Science!

This is the official implementation for the paper "Learning to Scaffold: Optimizing Model Explanations for Teaching"

The entmax mapping in JAX

A guide to building language technology in new languages.

Bicleaner is a parallel corpus classifier/cleaner that aims at detecting noisy sentence pairs in a parallel corpus.

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

A Neural Framework for MT Evaluation

A repository consisting of useful terminal commands required in daily tasks to reduce stackoverflow searches.

A workflow management system for researchers who heart Unix.

The Fast Cross-Platform Package Manager

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

Flax is a neural network library for JAX that is designed for flexibility.

AI Reading Group

A minimal C implementation of Nintendo Gameboy - An fast research environment for Reinforcement Learning

[CVPR 2020 Workshop] A PyTorch GAN library that reproduces research results for popular GANs.

TensorFlow implementations of Graph Neural Networks

Graph Neural Network Library for PyTorch

Replication of the paper "Structured Neural Summarization" which uses Graph Neural Networks and Seq2Seq models to summarize natural language and source code.

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Neural machine translation and sequence learning using TensorFlow

scikit-learn: machine learning in Python

A few data mining algorithms in pure python