Databricks Mosaic Research

All

7 repositories

llm-foundry
Public
LLM training code for Databricks foundation models
nlp deep-learning pytorch neural-networks llm
Python
•
Apache License 2.0
•537•4.1k•34•29•Updated Jan 8, 2025Jan 8, 2025
streaming
Public
A Data Streaming Library for Efficient Neural Network Training
streaming neural-network dataset machine-learning deep-learning pytorch
Python
•
Apache License 2.0
•148•1.2k•54•26•Updated Jan 7, 2025Jan 7, 2025
composer
Public
Supercharge Your Model Training
machine-learning deep-learning neural-network pytorch neural-networks ml-training ml-systems ml-efficiency
Python
•
Apache License 2.0
•431•5.2k•36•27•Updated Jan 6, 2025Jan 6, 2025
diffusion
Public
Python
•
Apache License 2.0
•71•684•18•4•Updated Nov 26, 2024Nov 26, 2024
ci-testing
Public
Aggregated CI test repo
0•1•0•0•Updated Nov 14, 2024Nov 14, 2024
examples
Public
Fast and flexible reference benchmarks
Shell
•
Apache License 2.0
•125•447•16•12•Updated Aug 14, 2024Aug 14, 2024
diffusion-benchmark
Public
Python
•
Apache License 2.0
•6•54•0•0•Updated Jan 16, 2024Jan 16, 2024