Stars
Solve puzzles. Improve your pytorch.
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
FFCV: Fast Forward Computer Vision (and other ML workloads!)
Ecosystem of libraries and tools for writing and executing fast GPU code fully in Rust.
A collection of tools for neural compression enthusiasts.
🐉 Making Rust a first-class language and ecosystem for GPU shaders 🚧
A data augmentations library for audio, image, text, and video.
A terminal workspace with batteries included
A comprehensive library for machine learning and numerical computing. Apply Machine Learning with Rust leveraging first principles.
Open Data Platform for analysts, quants and AI agents.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Statistics-driven benchmarking library for Rust
Research language for array processing in the Haskell/ML family
FSA/FST algorithms, differentiable, with PyTorch compatibility.
Benchmarks of approximate nearest neighbor libraries in Python
ClearML - Auto-Magical CI/CD to streamline your AI workload. Experiment Management, Data Management, Pipeline, Orchestration, Scheduling & Serving in one MLOps/LLMOps solution
ClearML - Auto-Magical CI/CD to streamline your AI workload. Experiment Management, Data Management, Pipeline, Orchestration, Scheduling & Serving in one MLOps/LLMOps solution
Code for the Eager Translation Model from the paper You May Not Need Attention
Dataset to assess the disentanglement properties of unsupervised learning methods
PyTorch implementation of "Emerging Disentanglement in Auto-Encoder Based Unsupervised Image Content Transfer"
Neural relational inference for interacting systems - pytorch
Implementation of "Learning Combinatorial Optimization Algorithms over Graphs"
A clone of the MiniPARSEC project inside the main PARSEC repository.