A python tool for evaluating the quality of sentence embeddings.

Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.

A framework for training tiny language models and studying their learning dynamics.

⏩ Continue is the leading open-source AI code assistant. You can connect any models and any context to build custom autocomplete and chat experiences inside VS Code and JetBrains

The nnsight package enables interpreting and manipulating the internals of deep learned models.

Next-Token Prediction is All You Need

Create feature-centric and prompt-centric visualizations for sparse autoencoders (like those from Anthropic's published research).

High accuracy RAG for answering questions from scientific documents with citations

Python Library for learning (Structure and Parameter), inference (Probabilistic and Causal), and simulations in Bayesian Networks.

An index of algorithms for learning causality with data

ALICE (Automated Learning and Intelligence for Causation and Economics) is a Microsoft Research project aimed at applying Artificial Intelligence concepts to economic decision making. One of its go…

Implement a ChatGPT-like LLM in PyTorch from scratch, step by step

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

Computations and statistics on manifolds with geometric structures.

A library for mechanistic interpretability of GPT-style language models

Python inventory optimization and simulation tools.

Causal Discovery in Python. It also includes (conditional) independence tests and score functions.

💻 Padrões de Projeto em Python

Perplexica is an AI-powered search engine. It is an Open source alternative to Perplexity AI

A framework for serving and evaluating LLM routers - save LLM costs without compromising quality

Kubernetes powered PaaS that runs in your own cloud.

Financial agent + generative UI

Platform to experiment with the AI Software Engineer. Terminal based. NOTE: Very different from https://gptengineer.app

We write your reusable computer vision tools. 💜

Machine Learning Toolkit for Kubernetes

End-to-end stack for WebRTC. SFU media server and SDKs.

🔥 Turn entire websites into LLM-ready markdown or structured data. Scrape, crawl and extract with a single API.

PyTorch native quantization and sparsity for training and inference

A flexible package for multimodal-deep-learning to combine tabular data with text and images using Wide and Deep models in Pytorch