Pytorch Developer Jobs in Prosper, TX (NOW HIRING)

We are building next-generation intelligent systems powered by AI and automation. Our team is looking for a skilled Python Developer for ML with deep expertise in framework design and exposure to Agentic AI systems. You will play a pivotal role in architecting scalable frameworks, APIs, and integrations while enabling AI-driven agents to perform complex workflows securely and efficiently.
 

Key Responsibilities:
Design, develop, and maintain Python frameworks that provide reusable components and structure for applications.
Build, train, and deploy robust machine learning and deep learning models using Python and popular frameworks (e.g., PyTorch, TensorFlow).
Architect modular, scalable, and extensible frameworks for APIs, data processing, or AI integrations.
Implement best practices in software engineering: OOP, design patterns, modular architecture.
Develop and optimize Retrieval-Augmented Generation (RAG) pipelines for Large Language Models (LLMs) to provide contextually relevant and accurate outputs. This includes ingesting, processing, and indexing data from various sources (e.g., PDFs, HTML, databases).
Develop APIs and SDKs for framework adoption across teams.
Utilize vector databases (e.g., Pinecone, Weaviate) for efficient storage and retrieval of high-dimensional vector embeddings to support RAG and semantic search functionalities.
Collaborate with AI engineers to integrate Agentic AI systems (e.g., AI agents, LLM orchestration frameworks like LangChain, LlamaIndex, Google ADK).
Ensure security, reliability, and observability of framework components.
Write unit/integration tests and maintain CI/CD pipelines.
Document frameworks, APIs, and libraries for developer adoption.
Participate in code reviews, mentor junior developers, and contribute to technical design discussions.
 

Required Skills & Experience:
Strong proficiency in Python (3.x) for ML and software development., with focus on OOP and modular design.
Proven experience in building frameworks, libraries, or SDKs (not just applications).
Advanced proficiency in Python for ML and software development.
Expertise in design patterns, dependency injection, and plugin-based architectures.
Solid understanding of ML algorithms, model evaluation techniques, and MLOps practices.
Experience developing with Large Language Models (LLMs) and advanced Prompt Engineering techniques.
Experience with FastAPI, Flask, Django or similar web frameworks.
Solid understanding of APIs (REST, GraphQL) and API gateway integration.
Knowledge of async programming (asyncio, aiohttp).
Hands-on experience building and deploying Retrieval-Augmented Generation (RAG) systems.
Proficiency in testing frameworks (PyTest, unittest) and CI/CD workflows.
Strong grasp of security practices (OAuth2, JWT, IAM integration).
Familiarity with cloud platforms (GCP, Azure, AWS) for deploying frameworks.
Familiarity with vector databases for efficient similarity search
 

Nice to Have:
Experience with Agentic AI frameworks (OpenAI SDK, LangChain, LlamaIndex, Google ADK, AutoGen).
Knowledge of LLM fine-tuning, prompt engineering, and guardrails.
Exposure to event-driven architectures (Kafka, Pub/Sub, Redis Streams).
Experience with observability tools (Prometheus, OpenTelemetry, Cloud Logging).
Contributions to open-source Python frameworks or libraries.
 

Ideal Candidate:
A framework builder mindset - you enjoy designing systems other developers will use.
Passionate about AI and automation, eager to experiment with agent-driven architectures.
A problem-solver with a strong understanding of scalable and secure design principles.
Excellent communicator with strong documentation and collaboration skills.
 

Qualifications
Bachelor's/Master's degree in Computer Science, Engineering, or related field.
Strong track record of building scalable frameworks or platforms in production environments.