Teddy Koker's starred repositories
Time-Series-Library
A Library for Advanced Deep Time Series Models.
llm-foundry
LLM training code for Databricks foundation models
sqlite-vec
A vector search SQLite extension that runs anywhere!
RFdiffusion
Code for running RFdiffusion
keras-core
A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.
cvxportfolio
Portfolio optimization and back-testing.
tensordict
TensorDict is a pytorch dedicated tensor container.
flash-attention-minimal
Flash Attention in ~100 lines of CUDA (forward pass only)
equiformer-pytorch
Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein folding
orb-models
ORB forcefield models from Orbital Materials
gateloop-transformer
Implementation of GateLoop Transformer in Pytorch and Jax
charge3net
Higher-order equivariant neural networks for charge density prediction in materials
MoonBoardRNN
A MoonBoard Problem Generator (DeepRouteSet) and Grade Predictor (GradeNet) based on LSTM