google-deepmind / xmanagerLinks
A platform for managing machine learning experiments
☆864Updated 2 weeks ago
Alternatives and similar repositories for xmanager
Users that are interested in xmanager are comparing it to the libraries listed below
Sorting:
- ML Collections is a library of Python Collections designed for ML use cases.☆968Updated 2 weeks ago
- ☆871Updated this week
- ☆350Updated this week
- A High Level API for Deep Learning in JAX☆476Updated 2 years ago
- PIX is an image processing library in JAX, for JAX.☆421Updated 4 months ago
- CLU lets you write beautiful training loops in JAX.☆350Updated last month
- A library for distributed ML training with PyTorch☆366Updated 2 years ago
- ☆771Updated last year
- ☆324Updated last year
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆522Updated this week
- ☆581Updated 3 weeks ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,435Updated this week
- A Graph Neural Network Library in Jax☆1,439Updated last year
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,438Updated 3 months ago
- ☆906Updated last year
- Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.☆1,584Updated this week
- 100 exercises to learn JAX☆587Updated 3 years ago
- Library for reading and processing ML training data.☆484Updated last week
- Optax is a gradient processing and optimization library for JAX.☆1,959Updated this week
- ☆782Updated last month
- Named tensors with first-class dimensions for PyTorch☆332Updated 2 years ago
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆480Updated 3 years ago
- A library to inspect and extract intermediate layers of PyTorch models.☆473Updated 3 years ago
- TensorDict is a pytorch dedicated tensor container.☆949Updated this week
- ☆361Updated last year
- High performance model preprocessing library on PyTorch☆649Updated last year
- maximal update parametrization (µP)☆1,576Updated last year
- JAX-based neural network library☆3,068Updated this week
- JAX - A curated list of resources https://github.com/google/jax☆1,883Updated 5 months ago
- Orbax provides common checkpointing and persistence utilities for JAX users☆410Updated this week