j3soon / tbparse
Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow
☆173Updated last month
Related projects: ⓘ
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- ☆149Updated 9 months ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆555Updated 4 months ago
- Code for our NeurIPS 2022 paper☆360Updated last year
- Aggregate multiple tensorboard runs to new summary or csv files☆164Updated 10 months ago
- Convert scikit-learn models to PyTorch modules☆157Updated 4 months ago
- TorchOpt is an efficient library for differentiable optimization built upon PyTorch.☆528Updated this week
- CLU lets you write beautiful training loops in JAX.☆319Updated 3 weeks ago
- JAX Synergistic Memory Inspector☆161Updated 2 months ago
- ☆104Updated last week
- Example of how to use Weights & Biases on Slurm☆108Updated 2 years ago
- A library to inspect and extract intermediate layers of PyTorch models.☆469Updated 2 years ago
- Reduce multiple PyTorch TensorBoard runs to new event (or CSV) files.☆68Updated 2 months ago
- Implementation of https://srush.github.io/annotated-s4☆457Updated last year
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆300Updated 6 months ago
- JMP is a Mixed Precision library for JAX.☆183Updated 3 months ago
- ☆290Updated 3 months ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆257Updated 10 months ago
- Pytorch Lightning Distributed Accelerators using Ray☆211Updated 10 months ago
- TensorDict is a pytorch dedicated tensor container.☆808Updated this week
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆199Updated last year
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆474Updated 2 years ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆176Updated last month
- ☆182Updated last year
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆213Updated this week
- ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning☆263Updated last year
- Use Jax functions in Pytorch☆224Updated last year
- Unofficial JAX implementations of deep learning research papers☆150Updated 2 years ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆176Updated 3 years ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆106Updated this week