Stars
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
An interactive NVIDIA-GPU process viewer and beyond, the one-stop solution for GPU process management.
Official Implementation of POYO-1 https://poyo-brain.github.io/
Ephys toolbox to match single units (electrophysiology) either within the same day (over splits) or across two days
Machine Learning for Alignment Bootcamp (MLAB).
Code / solutions for Mathematics for Machine Learning (MML Book)
A Lean Persistent Homology Library for Python
Modules for processing extracellular electrophysiology data from Neuropixels probes
Really Fast End-to-End Jax RL Implementations
Chrome extension that allows you to visualize equations using Latex notation inside ChatGPT
Extremely fast Query Engine for DataFrames, written in Rust
Python implementation of EM algorithm for GMM. And visualization for 2D case.
Neuropixels trajectory explorer with the Allen CCF mouse atlas
code to fit GLMs, HMMs, and GLM-HMMs (aka IO-HMMs)
Matlab interface for Latent Factor Analysis via Dynamical Systems (LFADS)
Project Aeon's "meta" repo that details the project, with a focus on dev practices and notes on technology review, challenges and insights.
Project Aeon's main library for interfacing with acquired data. Contains modules for raw data file io, data querying, data processing, data qc, database ingestion, and building computational data p…
Official implementation of DeepLabCut: Markerless pose estimation of user-defined features with deep learning for all animals incl. humans
Analysis of electrophysiological data recorded using Axona and openephys using Python
Bunch of utility functions that I use across a number of repos
Learnable latent embeddings for joint behavioral and neural analysis - Official implementation of CEBRA