keras-team / keras-core
A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.
☆1,269Updated last month
Alternatives and similar repositories for keras-core
Users that are interested in keras-core are comparing it to the libraries listed below
Sorting:
- Industry-strength Computer Vision workflows with Keras☆1,035Updated last month
- Pretrained model hub for Keras 3.☆896Updated this week
- 🤖 A PyTorch library of curated Transformer models and their composable components☆886Updated last year
- Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.☆1,560Updated last week
- Tensors, for human consumption☆1,249Updated 5 months ago
- Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)☆1,257Updated 5 months ago
- The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX☆418Updated last year
- TensorDict is a pytorch dedicated tensor container.☆925Updated this week
- Schedule-Free Optimization in PyTorch☆2,161Updated last month
- The WeightWatcher tool for predicting the accuracy of Deep Neural Networks☆1,592Updated 8 months ago
- Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackab…☆1,565Updated last year
- Thunder gives you PyTorch models superpowers for training and inference. Unlock out-of-the-box optimizations for performance, memory and …☆1,342Updated this week
- If tinygrad wasn't small enough for you...☆720Updated last year
- A simple, performant and scalable Jax LLM!☆1,725Updated this week
- A platform for managing machine learning experiments☆851Updated 2 weeks ago
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,407Updated 2 weeks ago
- 100 exercises to learn JAX☆578Updated 2 years ago
- Library for reading and processing ML training data.☆441Updated this week
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,779Updated 3 weeks ago
- Optax is a gradient processing and optimization library for JAX.☆1,891Updated this week
- For optimization algorithm research and development.☆513Updated this week
- Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimenta…☆496Updated 3 weeks ago
- Implementation of Hinton's forward-forward (FF) algorithm - an alternative to back-propagation☆1,480Updated last year
- IDE for PyTorch and its ecosystem☆388Updated last year
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,432Updated 2 weeks ago
- What would you do with 1000 H100s...☆1,045Updated last year
- maximal update parametrization (µP)☆1,511Updated 10 months ago
- Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.☆1,056Updated 3 weeks ago
- skops is a Python library helping you share your scikit-learn based models and put them in production☆480Updated 3 weeks ago
- Puzzles for learning Triton☆1,623Updated 6 months ago