pytorch / pytorch_sphinx_theme
PyTorch Sphinx Theme
☆96Updated 6 months ago
Related projects ⓘ
Alternatives and complementary repositories for pytorch_sphinx_theme
- Configuration classes enabling type-safe PyTorch configuration for Hydra apps☆205Updated 2 years ago
- Automate issue discovery for your projects against Lightning nightly and releases.☆45Updated this week
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆253Updated 2 years ago
- Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.☆50Updated last month
- pytest plugin for a better developer experience when working with the PyTorch test suite☆43Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated last year
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆126Updated 10 months ago
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆73Updated 11 months ago
- Annotating tensor shapes using Python types☆158Updated 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…☆216Updated last week
- Multidimensional indexing for tensors☆113Updated last year
- JMP is a Mixed Precision library for JAX.☆187Updated 6 months ago
- Torch Distributed Experimental☆116Updated 3 months ago
- Named tensors with first-class dimensions for PyTorch☆322Updated last year
- Configuration classes enabling Hydra to configure and manage Pytorch Lightning projects.☆40Updated 3 years ago
- A collection of code snippets for my PyTorch Lightning projects☆107Updated 3 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 3 years ago
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆175Updated last week
- NumPy-style histograms in PyTorch☆51Updated 7 months ago
- Collection of eclectic utils for python.☆193Updated this week
- Documentation:☆120Updated last year
- Public repository for managing Grid Platform documentation synced with gitbook on docs.grid.ai☆20Updated 2 years ago
- RFC document, tooling and other content related to the array API standard☆215Updated 3 weeks ago
- Pytorch Lightning Distributed Accelerators using Ray☆211Updated last year
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- CLU lets you write beautiful training loops in JAX.☆321Updated this week
- ☆105Updated 2 weeks ago
- NumPy arrays, ready for human consumption☆63Updated 4 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆102Updated 4 months ago
- Placeholder for the opensource Grid AI components☆44Updated 2 years ago