Skip to content

@layer6ai-labs Layer 6 Labs

Change the repository type filter

All

    Repositories list

    65 repositories

    • LoCalPFN

      Public
      Retrieval & Fine-Tuning for In-Context Tabular Models (NeurIPS 2024)
      Python
      •0•1•0•0•Updated Nov 7, 2025Nov 7, 2025

    • ssrl-mi-maximization

      Public
      Self-Supervised Representation Learning as Mutual Information Maximization
      Jupyter Notebook
      •
      MIT License
      •0•1•0•0•Updated Nov 5, 2025Nov 5, 2025

    • conformal-rag

      Public
      Assessing Quality of Retrieval-Augmented Generation via Conditional Conformal Factuality
      Python
      •
      MIT License
      •0•1•0•0•Updated Nov 5, 2025Nov 5, 2025

    • CausalPFN

      Public
      CausalPFN: Amortized Causal Effect Estimation via In-Context Learning
      Jupyter Notebook
      •
      Apache License 2.0
      •0•1•0•0•Updated Nov 5, 2025Nov 5, 2025

    • DataCVChallenge-2024-

      Public
      Repository for our code for DataCV Challenge in CVPR 2024
      0•0•0•0•Updated Nov 4, 2025Nov 4, 2025

    • act_ids

      Public
      Python
      •0•0•0•0•Updated Nov 4, 2025Nov 4, 2025

    • rag-error-classification

      Public
      Classifying errors in RAG systems to assist iterative model refinement
      Python
      •
      MIT License
      •0•3•1•0•Updated Oct 20, 2025Oct 20, 2025

    • conversational-txt2sql

      Public
      Python
      •
      MIT License
      •1•0•0•0•Updated Oct 18, 2025Oct 18, 2025

    • TabDPT-inference

      Public
      Inference code for "TabDPT: Scaling Tabular Foundation Models on Real Data"
      Python
      •
      Apache License 2.0
      •12•60•4•2•Updated Oct 16, 2025Oct 16, 2025

    • conformal-importance-summarization

      Public
      Document summarization with conformal guarantees on information retention.
      Python
      •
      MIT License
      •0•1•1•0•Updated Sep 25, 2025Sep 25, 2025

    • DRESS

      Public
      The code base for the research project exploring disentangled representation-based self-supervised meta-learning.
      Python
      •
      MIT License
      •0•5•2•0•Updated Sep 25, 2025Sep 25, 2025

    • TabDPT-training

      Public
      Training code for TabDPT: Scaling Tabular Foundation Models on Real Data
      Python
      •
      Other
      •2•37•0•2•Updated Aug 3, 2025Aug 3, 2025

    • flipd

      Public
      An efficient local intrinsic dimension estimator with diffusion models.
      Jupyter Notebook
      •1•2•2•0•Updated Jun 11, 2025Jun 11, 2025

    • diffusion_memorization

      Public
      Jupyter Notebook
      •1•2•0•0•Updated May 20, 2025May 20, 2025

    • spark-retrieval-tuning

      Public
      C
      •
      MIT License
      •0•4•0•0•Updated May 9, 2025May 9, 2025

    • last_layer_empirical_bayes

      Public
      0•1•0•0•Updated Mar 5, 2025Mar 5, 2025

    • dgm-eval

      Public
      Codebase for evaluation of deep generative models as presented in Exposing flaws of generative model evaluation metrics and their unfair treatment of diffusion models
      Jupyter Notebook
      •
      MIT License
      •18•195•3•0•Updated Mar 3, 2025Mar 3, 2025

    • msc-sql

      Public
      Text-2-SQL
      Python
      •
      MIT License
      •2•20•1•0•Updated Feb 21, 2025Feb 21, 2025

    • conformal-prediction-fairness

      Public
      Evaluating fairness of conformal prediction with human interaction
      Python
      •
      MIT License
      •1•4•0•0•Updated Feb 13, 2025Feb 13, 2025

    • dgm_geometry

      Public
      Jupyter Notebook
      •
      MIT License
      •3•16•0•0•Updated Jan 20, 2025Jan 20, 2025

    • direct-cms

      Public
      Inconsistencies In Consistency Models: Better ODE Solving Does Not Imply Better Samples
      Python
      •0•4•0•0•Updated Nov 15, 2024Nov 15, 2024

    • calo-forest

      Public
      A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.
      Python
      •
      MIT License
      •1•18•0•0•Updated Nov 3, 2024Nov 3, 2024

    • calo-man

      Public
      Manifold-aware deep generative models for calorimeter simulation.
      Python
      •0•2•0•0•Updated Sep 29, 2024Sep 29, 2024

    • dgm_manifold_survey

      Public
      Python
      •0•4•0•0•Updated Sep 24, 2024Sep 24, 2024

    • two_step_zoo

      Public
      Codebase accompanying the paper "Diagnosing and Fixing Manifold Overfitting in Deep Generative Models"
      Jupyter Notebook
      •
      Apache License 2.0
      •2•9•0•0•Updated Aug 26, 2024Aug 26, 2024

    • dgm_ood_detection

      Public
      Code for the paper "A Geometric Explanation of the Likelihood OOD Detection Paradox"
      Jupyter Notebook
      •
      MIT License
      •0•3•0•0•Updated Jul 18, 2024Jul 18, 2024

    • ssl-robustness

      Public
      Codebase for benchmarking robustness of Self-Supervised Learning across diverse downstream tasks
      Python
      •
      MIT License
      •0•6•0•0•Updated Jul 18, 2024Jul 18, 2024

    • tpcds-spark

      Public
      C
      •
      MIT License
      •1•0•0•0•Updated Jul 17, 2024Jul 17, 2024

    • dp-cert

      Public
      A framework for providing certified robustness guarantees with differentially private models
      Jupyter Notebook
      •
      MIT License
      •1•1•0•0•Updated Jul 16, 2024Jul 16, 2024

    • hitl-conformal-prediction

      Public
      Human in the Loop Conformal Prediction
      Python
      •
      MIT License
      •2•6•0•0•Updated May 12, 2024May 12, 2024