YNCris

MetaTransformer

Meta-Transformer for Unified Multimodal Learning

CellBox

CellBox: Interpretable Machine Learning for Perturbation Biology

alphafold

Open source code for AlphaFold.

dm_control

Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.

scAnt

open-source 3D scanning and processing pipeline

replicAnt

replicAnt - generating annotated images of animals in complex environments with Unreal Engine

cleanrl

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

gym

A toolkit for developing and comparing reinforcement learning algorithms.

Gymnasium

An API standard for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)

taming-transformers

Taming Transformers for High-Resolution Image Synthesis

stable-diffusion

A latent text-to-image diffusion model

stable-diffusion-webui

Stable Diffusion web UI

habitat-lab

A modular high-level library to train embodied AI agents across a variety of tasks and environments.

Pytorch-code-for-time-series-classification

Pytorch code for mutil-channel time series dataset. You can use this project to train LSTM to classify such data.

Batch-Channel-Normalization

SBeA_release

The release version of Social Behavior Atlas (SBeA)

flygym

Gym environments for NeuroMechFly in various physics simulators

NeuroMechFly

A neuromechanical model of adult Drosophila melanogaster.

ActCLR

keypoint-moseq

ADPT

Anti-drift pose tracker (ADPT): A transformer-based network for robust animal pose estimation cross-species

Sparse-SIM

Official MATLAB implementation of the "Sparse deconvolution" -v1.0.3

sacrebleu

Reference BLEU implementation that auto-downloads test sets and reports a version string to facilitate cross-lab comparisons

FastSAM

Fast Segment Anything

CEBRA

Learnable latent embeddings for joint behavioral and neural analysis - Official implementation of CEBRA

electrophysiology

mmpose

OpenMMLab Pose Estimation Toolbox and Benchmark.

annotated-transformer

An annotated implementation of the Transformer paper.

segment-anything

The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.

transformer-models

Deep Learning Transformer models in MATLAB