Lightning-AI / LitLoggerLinks
A minimal Python logger that tracks everything you try when building AI - metrics, prompts, models, etc, so you can see what changed and why.
☆16Updated this week
Alternatives and similar repositories for LitLogger
Users that are interested in LitLogger are comparing it to the libraries listed below
Sorting:
- TensorDict is a pytorch dedicated tensor container.☆1,003Updated last week
- Speed up model training by fixing data loading.☆575Updated this week
- PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily wri…☆1,439Updated this week
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,247Updated this week
- PyTorch Single Controller☆957Updated this week
- Scalable and Performant Data Loading☆364Updated this week
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆474Updated 3 weeks ago
- PyTorch native quantization and sparsity for training and inference☆2,657Updated last week
- Save, load, host, and share AI model checkpoints without slowing down training. Host on Lightning AI or your own cloud with enterprise-gr…☆42Updated this week
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆246Updated 3 weeks ago
- For optimization algorithm research and development.☆558Updated 3 weeks ago
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,072Updated last year
- Helpful tools and examples for working with flex-attention☆1,118Updated 3 weeks ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆596Updated 5 months ago
- A lightweight, local-first, and 🆓 experiment tracking library from Hugging Face 🤗☆1,244Updated last week
- Library for reading and processing ML training data.☆677Updated last week
- Best practices & guides on how to write distributed pytorch training code☆575Updated 3 months ago
- ML Collections is a library of Python Collections designed for ML use cases.☆1,012Updated last week
- torchview: visualize pytorch models☆1,029Updated 8 months ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆739Updated this week
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,723Updated last week
- A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.☆1,266Updated 8 months ago
- A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.☆921Updated this week
- Orbax provides common checkpointing and persistence utilities for JAX users☆479Updated this week
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,058Updated 5 months ago
- The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX☆423Updated last year
- ☆558Updated last year
- TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.☆1,009Updated last week
- A library to analyze PyTorch traces.☆462Updated this week
- Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.☆636Updated 4 months ago