Databricks Mosaic Research

All

8 repositories

llm-foundry
Public
LLM training code for Databricks foundation models
nlp deep-learning pytorch neural-networks llm
Python
•
Apache License 2.0
•573•4.3k•40•15•Updated Jul 23, 2025Jul 23, 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
•449•5.4k•35•14•Updated Jul 23, 2025Jul 23, 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
•165•1.3k•69•27•Updated Jul 17, 2025Jul 17, 2025
examples
Public
Fast and flexible reference benchmarks
Shell
•
Apache License 2.0
•128•462•16•12•Updated Jun 23, 2025Jun 23, 2025
ci-testing
Public
Aggregated CI test repo
0•1•0•0•Updated Jun 2, 2025Jun 2, 2025
docs
Public
Python
•
Other
•0•0•0•0•Updated Jan 17, 2025Jan 17, 2025
diffusion
Public
Python
•
Apache License 2.0
•73•704•18•4•Updated Jan 10, 2025Jan 10, 2025
diffusion-benchmark
Public
Python
•
Apache License 2.0
•6•54•0•0•Updated Jan 16, 2024Jan 16, 2024