Accelerate, Optimize performance with streamlined training and serving options with JAX.

JAX-Toolbox

A flexible and efficient implementation of Flash Attention 2.0 for JAX, supporting multiple backends (GPU/TPU/CPU) and platforms (Triton/Pallas/JAX).

Simple benchmarks of transformers comparing 3090 with Titan RTX

Python logging made (stupidly) simple

Aperol is a featherweight framework for configuration of Python objects

Everything you want to know about Google Cloud TPU

A functional training loops library for JAX

Convert Machine Learning Code Between Frameworks

JAX Synergistic Memory Inspector

Robust Speech Recognition via Large-Scale Weak Supervision

Accelerated Quality-Diversity

MMseqs2: ultra fast and sensitive search and clustering suite

Lightweight, super fast C/C++ (& Python) library for sequence alignment using edit (Levenshtein) distance.

Fastest-lap is a vehicle dynamics simulator. It can be used to understand vehicle dynamics, to learn about driving techniques, to design car prototypes, or just for fun!

Github for files currently published in the IPD-IMGT/HLA FTP Directory hosted at the European Bioinformatics Institute

functorch is JAX-like composable function transforms for PyTorch.

Central place for the engineering/scaling WG: documentation, SLURM scripts and logs, compute environment and data.

An open access book on scientific visualization using python and matplotlib

Code for "T Cell Receptor Specificity Prediction with Bimodal Attention Networks" (https://doi.org/10.1093/bioinformatics/btab294, ISMB 2021)

PaccMann models for protein language modeling

FUSE-based file system backed by Amazon S3

Parse Python docstrings in various flavors.

Implement minimal boilerplate CLIs derived from type hints and parse from command line, config files and environment variables

Listing of papers about machine learning for proteins.

Ultimate Time Tracker - A simple command-line time tracker written in Python

Python and tab completion, better together.