Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
⚙️ Bibliotecas de deep learning
PyTorch, Transformers, JAX e outros frameworks e ferramentas de baixo nível
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Machine learning metrics for distributed, scalable PyTorch applications.
A simple, performant, and scalable Jax LLM!
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
A Julia machine learning framework
Automated Machine Learning on Kubernetes
An acausal modeling framework for automatically parallelized scientific machine learning (SciML) in Julia. A computer algebra system for integrated symbolics for physics-informed machine learning and automated transforma
A model-agnostic visual debugging tool for machine learning
A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.