umangjpatel / keraxLinks
Keras-like APIs for JAX framework
☆50Updated 2 years ago
Alternatives and similar repositories for kerax
Users that are interested in kerax are comparing it to the libraries listed below
Sorting:
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- A queue service for quickly developing scripts that use all your GPUs efficiently☆88Updated 2 years ago
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago
- Image augmentation library for Jax☆39Updated last year
- ☆31Updated 2 years ago
- Functional machine learning for fun☆85Updated 4 years ago
- Documentation:☆121Updated 2 years ago
- A lightweight library for tensorflow 2.0☆66Updated 5 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated last month
- Draw, modify, and execute graphs using a simple python dict interface☆31Updated 2 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- Experiment orchestration☆102Updated 5 years ago
- Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.☆57Updated 3 years ago
- TF 2.x and PyTorch Lightning Callbacks for GPU monitoring☆92Updated 5 years ago
- Quick modules to turn regular Neural Networks to Bayesian Neural Networks with Dropout.☆35Updated 4 years ago
- High Performance Tensorflow Data Pipeline with State of Art Augmentations and low level optimizations.☆85Updated 3 years ago
- Version control for software 2.0☆65Updated 4 years ago
- Blazingly fast capsule networks in 75 lines of pytorch+einops☆26Updated 3 years ago
- Composable kernels for scikit-learn implemented in JAX.☆44Updated 4 years ago
- Texture mapping with variational auto-encoders☆40Updated 3 years ago
- A metrics library for the JAX ecosystem☆40Updated 2 years ago
- Web Application to generate your training scripts with PyTorch Ignite☆44Updated this week
- notebooks of cool EBM visualizations☆15Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆112Updated 3 years ago
- A simple wrapper over `pydot` and `graphviz` which fixes some sharp edges☆63Updated 2 years ago
- Automate issue discovery for your projects against Lightning nightly and releases.☆46Updated 3 months ago
- An attempt at the implementation of GLOM, Geoffrey Hinton's paper for emergent part-whole hierarchies from data☆37Updated 4 years ago
- ☆69Updated last year
- PyTorch model training and layer saturation monitor☆82Updated 2 years ago
- NeuralPy: A Keras like deep learning library works on top of PyTorch☆79Updated last year