Optax is a gradient processing and optimization library for JAX
A simple but complete full-attention transformer
Images to inference with no labeling
A library for accelerating Transformer models on NVIDIA GPUs
Book about interpretable machine learning
Making Enterprise Data Intelligent and Responsive for AI
Trainable models and NN optimization tools
Decentralized deep learning in PyTorch. Built to train models
A comprehensive set of fairness metrics for datasets
Data loaders and abstractions for text and NLP
Streamline your ML workflow
Probabilistic reasoning and statistical analysis in TensorFlow
Open deep learning compiler stack for cpu, gpu, etc.
Geometric deep learning extension library for PyTorch
Lightning fast C++/CUDA neural network framework
Solve end to end problems using Llama model family
A TensorFlow library for applied reinforcement learning
Create HTML profiling reports from pandas DataFrame objects
Deep learning optimization library making distributed training easy
High-performance library for gradient boosting on decision trees
Detecting silent model failure. NannyML estimates performance
PyTorch extensions for fast R&D prototyping and Kaggle farming
Spatiotemporal Signal Processing with Neural Machine Learning Models
Data science on data without acquiring a copy